| /frameworks/ml/nn/common/operations/ |
| D | LocalResponseNormalization.cpp | 35 constexpr uint32_t kNumInputs = 6; variable
123 int32_t axis = context->getNumInputs() == kNumInputs in executeTyped()
137 NN_RET_CHECK(context->getNumInputs() == kNumInputs || in validate()
138 context->getNumInputs() == kNumInputs - 1); in validate()
162 if (context->getNumInputs() == kNumInputs) { in validate()
180 int32_t axis = context->getNumInputs() == kNumInputs in prepare()
|
| D | ResizeImageOps.cpp | 37 constexpr uint32_t kNumInputs = 4; variable
177 NN_RET_CHECK(numInputs >= kNumInputs - 1 && numInputs <= kNumInputs + kNumOptionalInputs); in validate()
179 NN_RET_CHECK(numInputs >= kNumInputs && numInputs <= kNumInputs + kNumOptionalInputs); in validate()
209 if (numInputs < kNumInputs) { in validate()
211 } else if (numInputs == kNumInputs) { in validate()
|
| D | LogicalNot.cpp | 27 constexpr uint32_t kNumInputs = 1; variable
48 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Rank.cpp | 28 constexpr uint32_t kNumInputs = 1; variable
35 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Elementwise.cpp | 30 constexpr uint32_t kNumInputs = 1; variable
89 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
101 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateAbs()
114 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateFloor()
|
| D | L2Normalization.cpp | 36 constexpr uint32_t kNumInputs = 2; variable
203 NN_RET_CHECK(context->getNumInputs() == kNumInputs || in validate()
204 context->getNumInputs() == kNumInputs - 1); in validate()
218 if (context->getNumInputs() == kNumInputs) { in validate()
235 int32_t axis = context->getNumInputs() == kNumInputs in prepare()
258 int32_t axis = context->getNumInputs() == kNumInputs in execute()
|
| D | Elu.cpp | 35 constexpr uint32_t kNumInputs = 2; variable
59 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Neg.cpp | 32 constexpr uint32_t kNumInputs = 1; variable
54 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Softmax.cpp | 38 constexpr uint32_t kNumInputs = 3; variable
232 NN_RET_CHECK(context->getNumInputs() == kNumInputs || in validate()
233 context->getNumInputs() == kNumInputs - 1); in validate()
253 if (context->getNumInputs() == kNumInputs) { in validate()
280 int32_t axis = (context->getNumInputs() == kNumInputs) in execute()
|
| D | Fill.cpp | 27 constexpr uint32_t kNumInputs = 2; variable
68 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Reduce.cpp | 34 constexpr uint32_t kNumInputs = 3; variable
73 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateProdSum()
90 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateMaxMin()
113 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateLogical()
|
| D | LogicalAndOr.cpp | 28 constexpr uint32_t kNumInputs = 2; variable
64 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Squeeze.cpp | 32 constexpr uint32_t kNumInputs = 2; variable
42 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Quantize.cpp | 32 constexpr uint32_t kNumInputs = 1; variable
70 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | LogSoftmax.cpp | 34 constexpr uint32_t kNumInputs = 3; variable
75 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Select.cpp | 28 constexpr uint32_t kNumInputs = 3; variable
73 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | TopK_V2.cpp | 31 constexpr uint32_t kNumInputs = 2; variable
80 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | ChannelShuffle.cpp | 32 constexpr uint32_t kNumInputs = 3; variable
64 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Transpose.cpp | 36 constexpr uint32_t kNumInputs = 2; variable
77 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Dequantize.cpp | 28 constexpr uint32_t kNumInputs = 1; variable
80 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | InstanceNormalization.cpp | 33 constexpr uint32_t kNumInputs = 5; variable
106 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Gather.cpp | 30 constexpr uint32_t kNumInputs = 3; variable
66 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | PRelu.cpp | 38 constexpr uint32_t kNumInputs = 2; variable
102 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| D | Slice.cpp | 32 constexpr uint32_t kNumInputs = 3; variable
85 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
|
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciDriver.cpp | 45 constexpr uint32_t kNumInputs = 1; variable
68 NN_RET_CHECK_EQ(operation.inputs.size(), kNumInputs); in validate()
|