Home
last modified time | relevance | path

Searched refs:kNumInputs (Results 1 – 25 of 38) sorted by relevance

12

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLocalResponseNormalization.cpp37 constexpr uint32_t kNumInputs = 6; variable
125 int32_t axis = context->getNumInputs() == kNumInputs in executeTyped()
140 NN_RET_CHECK(context->getNumInputs() == kNumInputs || in validate()
141 context->getNumInputs() == kNumInputs - 1); in validate()
166 if (context->getNumInputs() == kNumInputs) { in validate()
186 int32_t axis = context->getNumInputs() == kNumInputs in prepare()
H A DResizeImageOps.cpp38 constexpr uint32_t kNumInputs = 4; variable
180 NN_RET_CHECK(numInputs >= kNumInputs - 1 && numInputs <= kNumInputs + kNumOptionalInputs); in validate()
182 NN_RET_CHECK(numInputs >= kNumInputs && numInputs <= kNumInputs + kNumOptionalInputs); in validate()
213 if (numInputs < kNumInputs) { in validate()
215 } else if (numInputs == kNumInputs) { in validate()
H A DL2Normalization.cpp38 constexpr uint32_t kNumInputs = 2; variable
205 NN_RET_CHECK(context->getNumInputs() == kNumInputs || in validate()
206 context->getNumInputs() == kNumInputs - 1); in validate()
221 if (context->getNumInputs() == kNumInputs) { in validate()
240 int32_t axis = context->getNumInputs() == kNumInputs in prepare()
263 int32_t axis = context->getNumInputs() == kNumInputs in execute()
H A DElementwise.cpp29 constexpr uint32_t kNumInputs = 1; variable
86 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
98 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateAbs()
110 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateFloor()
H A DLogicalNot.cpp26 constexpr uint32_t kNumInputs = 1; variable
45 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DRank.cpp27 constexpr uint32_t kNumInputs = 1; variable
34 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DSoftmax.cpp42 constexpr uint32_t kNumInputs = 3; variable
236 NN_RET_CHECK(context->getNumInputs() == kNumInputs || in validate()
237 context->getNumInputs() == kNumInputs - 1); in validate()
258 if (context->getNumInputs() == kNumInputs) { in validate()
287 int32_t axis = (context->getNumInputs() == kNumInputs) in execute()
H A DNeg.cpp31 constexpr uint32_t kNumInputs = 1; variable
51 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DReduce.cpp35 constexpr uint32_t kNumInputs = 3; variable
74 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateProdSum()
91 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateMaxMin()
114 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateLogical()
H A DElu.cpp32 constexpr uint32_t kNumInputs = 2; variable
56 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DFill.cpp26 constexpr uint32_t kNumInputs = 2; variable
65 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DLogicalAndOr.cpp30 constexpr uint32_t kNumInputs = 2; variable
64 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DQuantize.cpp31 constexpr uint32_t kNumInputs = 1; variable
67 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DSelect.cpp27 constexpr uint32_t kNumInputs = 3; variable
70 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DSqueeze.cpp31 constexpr uint32_t kNumInputs = 2; variable
39 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DLogSoftmax.cpp33 constexpr uint32_t kNumInputs = 3; variable
74 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DChannelShuffle.cpp29 constexpr uint32_t kNumInputs = 3; variable
61 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DTopK_V2.cpp30 constexpr uint32_t kNumInputs = 2; variable
77 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DDequantize.cpp27 constexpr uint32_t kNumInputs = 1; variable
79 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DGather.cpp29 constexpr uint32_t kNumInputs = 3; variable
63 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DInstanceNormalization.cpp35 constexpr uint32_t kNumInputs = 5; variable
108 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DPRelu.cpp37 constexpr uint32_t kNumInputs = 2; variable
103 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DTranspose.cpp37 constexpr uint32_t kNumInputs = 2; variable
78 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
H A DSlice.cpp34 constexpr uint32_t kNumInputs = 3; variable
87 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
/aosp12/packages/modules/NeuralNetworks/runtime/test/fibonacci_extension/
H A DFibonacciDriver.cpp44 constexpr uint32_t kNumInputs = 1; variable
67 NN_RET_CHECK_EQ(operation.inputs.size(), kNumInputs); in validate()

12