Searched refs:numInputs (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | ResizeImageOps.cpp | 178 const auto numInputs = context->getNumInputs(); in validate() local
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()
219 while (inExpectedTypes.size() < numInputs) { in validate()
233 const auto numInputs = context->getNumInputs(); in prepare() local
|
| D | Concatenation.cpp | 183 uint32_t numInputs = context->getNumInputs(); in prepare() local
184 NN_RET_CHECK_GE(numInputs, 2); in prepare()
187 int32_t axis = context->getInputValue<int32_t>(numInputs - 1); in prepare()
193 for (uint32_t i = 1; i < numInputs - 1; ++i) { in prepare()
|
| D | DepthwiseConv2D.cpp | 422 const uint32_t numInputs = context->getNumInputs(); in validate() local
424 std::binary_search(std::begin(kNumInputsArray), std::end(kNumInputsArray), numInputs)); in validate()
481 if (numInputs >= 9) { in validate()
482 if (context->getInputType(8) == OperandType::INT32 && numInputs >= 11) { in validate()
489 if (numInputs >= 9 + inputOffset) { in validate()
493 NN_RET_CHECK_NE(numInputs, 10 + inputOffset) in validate()
496 if (numInputs == 11 + inputOffset) { in validate()
|
| D | Conv2D.cpp | 535 const uint32_t numInputs = context->getNumInputs(); in validate() local
537 std::binary_search(std::begin(kNumInputsArray), std::end(kNumInputsArray), numInputs)); in validate()
|
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| D | ConcatSplit.cpp | 24 static void concatConstructor(uint32_t numInputs, bool isV1_0, uint32_t rank, RandomOperation* op) { in concatConstructor() argument
25 for (uint32_t i = 0; i < numInputs; i++) { in concatConstructor()
32 op->inputs[numInputs]->setScalarValue<int32_t>(axis); in concatConstructor()
36 for (uint32_t j = 1; j < numInputs; j++) { in concatConstructor()
|
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/ |
| D | RandomGraphGenerator.cpp | 185 uint32_t numInputs = 0; in generateValue() local
187 if (operand->type == RandomOperandType::INPUT) numInputs++; in generateValue()
198 if (asConstant(operand, prob) && numInputs > 1) { in generateValue()
199 if (operand->type == RandomOperandType::INPUT) numInputs--; in generateValue()
|