Home
last modified time | relevance | path

Searched refs:inputCount (Results 1 – 24 of 24) sorted by relevance

/packages/modules/NeuralNetworks/common/
DLegacyUtils.cpp158 OperationValidationContext(const char* operationName, uint32_t inputCount, in OperationValidationContext() argument
162 inputCount(inputCount), in OperationValidationContext()
184 uint32_t inputCount; member in android::nn::__anon0471dca30111::OperationValidationContext
196 CHECK(index < static_cast<uint32_t>(inputCount)); in getInputOperand()
206 return inputCount; in getNumInputs()
585 static bool validateIfOperation(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, in validateIfOperation() argument
590 NN_RET_CHECK_GE(inputCount, 3u) << "ANEURALNETWORKS_IF must have at least 3 inputs"; in validateIfOperation()
597 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + branchModelInputCount); in validateIfOperation()
628 static bool validateWhileOperation(uint32_t inputCount, const uint32_t* inputs, in validateWhileOperation() argument
642 NN_RET_CHECK_GE(inputCount, 3u) << "ANEURALNETWORKS_WHILE must have at least 3 inputs"; in validateWhileOperation()
[all …]
DLegacyHalUtils.cpp204 static inline int validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount, in validateOperation() argument
213 return validateOperation(opType, inputCount, inputIndexes, outputCount, outputIndexes, operands, in validateOperation()
/packages/modules/NeuralNetworks/runtime/
DModelBuilder.cpp362 int ModelBuilder::addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, in addOperation() argument
401 NN_VALIDATE_NULL_OR_SIZED("addOperation", inputs, inputCount); in addOperation()
405 .inputs = makeVector(inputs, inputCount), in addOperation()
430 int ModelBuilder::identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, in identifyInputsAndOutputs() argument
436 NN_VALIDATE_NULL_OR_SIZED("identifyInputsAndOutputs", inputs, inputCount); in identifyInputsAndOutputs()
437 if (auto result = validateOperandList(makeVector(inputs, inputCount), operandCount(), in identifyInputsAndOutputs()
480 if (!setArguments(&mInputIndexes, inputCount, inputs, Operand::LifeTime::SUBGRAPH_INPUT) || in identifyInputsAndOutputs()
593 const uint32_t inputCount = operation.inputs.size(); in removeTrailingArgumentsWithDefaultValues() local
594 CHECK_LT(count, inputCount); in removeTrailingArgumentsWithDefaultValues()
595 const uint32_t newInputCount = inputCount - count; in removeTrailingArgumentsWithDefaultValues()
[all …]
DModelBuilder.h55 int addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, const uint32_t* inputs,
57 int identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount,
87 uint32_t inputCount() const { return static_cast<uint32_t>(mInputIndexes.size()); } in inputCount() function
DCompilationBuilder.cpp291 if (index >= mModel->inputCount()) { in getPreferredMemoryAlignmentForInput()
313 if (index >= mModel->inputCount()) { in getPreferredMemoryPaddingForInput()
414 if (index >= mModel->inputCount()) { in forEachStepRoleOfInput()
DExecutionPlan.cpp579 const uint32_t inputCount = static_cast<uint32_t>(operation.inputs.size()); in addOperation() local
581 std::vector<uint32_t> inputs(inputCount); in addOperation()
585 return mStepModel.addOperation(static_cast<uint32_t>(operation.type), inputCount, inputs.data(), in addOperation()
820 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finishStepModel()
993 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finish()
2043 CHECK_LT(index, mainModel->inputCount()); in getInputSourceOperand()
2312 for (uint32_t i = 0, n = thenModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2321 for (uint32_t i = 0, n = elseModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2373 for (uint32_t i = 0, n = condModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2380 for (uint32_t i = 0, n = bodyModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
DTelemetry.cpp106 for (size_t i = 0; i < m->inputCount(); i++) { in evalInputDataClass()
DNeuralNetworks.cpp1166 ANeuralNetworksOperationType type, uint32_t inputCount, in ANeuralNetworksModel_addOperation() argument
1175 return m->addOperation(type, inputCount, inputs, outputCount, outputs); in ANeuralNetworksModel_addOperation()
1191 int ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount, in ANeuralNetworksModel_identifyInputsAndOutputs() argument
1200 return m->identifyInputsAndOutputs(inputCount, inputs, outputCount, outputs); in ANeuralNetworksModel_identifyInputsAndOutputs()
DExecutionBuilder.cpp158 mInputs(mModel->inputCount()), in ExecutionBuilder()
1370 mInputs(model->inputCount()), in StepExecutor()
DMemory.cpp347 if (index >= model->inputCount()) { in addRole()
/packages/modules/NeuralNetworks/common/types/operations/src/
DPooling.cpp29 auto inputCount = context->getNumInputs(); in validate() local
30 NN_RET_CHECK(inputCount == 11 || inputCount == 10 || inputCount == 8 || inputCount == 7); in validate()
75 if (inputCount >= 10) { in validate()
80 if (inputCount == 11 || inputCount == 8) { in validate()
DConcatenation.cpp28 uint32_t inputCount = context->getNumInputs(); in validate() local
29 NN_RET_CHECK_GE(inputCount, 2u); in validate()
42 std::vector<OperandType> inExpectedTypes(inputCount - 1, inputType); in validate()
46 for (uint32_t i = 0; i < inputCount - 1; ++i) { in validate()
53 for (uint32_t i = 0; i < inputCount - 1; ++i) { in validate()
DConv2D.cpp39 auto inputCount = context->getNumInputs(); in validate() local
90 if (inputCount >= 8) { in validate()
91 if (context->getInputType(7) == OperandType::INT32 && inputCount >= 10) { in validate()
98 if (inputCount >= 8u + inputOffset) { in validate()
102 NN_RET_CHECK_NE(inputCount, 9u + inputOffset) in validate()
105 if (inputCount == 10u + inputOffset) { in validate()
DDensify.cpp28 const uint32_t inputCount = context->getNumInputs(); in validate() local
29 NN_RET_CHECK_GE(inputCount, kMinNumInputs); in validate()
30 NN_RET_CHECK_EQ(inputCount, in validate()
34 for (uint32_t i = 1; i < inputCount; i++) { in validate()
DTransposeConv2D.cpp27 const uint32_t inputCount = context->getNumInputs(); in validate() local
28 NN_RET_CHECK(inputCount == kNumInputs1 || inputCount == kNumInputs2); in validate()
58 if (inputCount == 11) { in validate()
/packages/modules/NeuralNetworks/common/cpu_operations/
DConcatenation.cpp99 uint32_t inputCount = context->getNumInputs() - 1; in concatenation() local
102 for (uint32_t i = 0; i < inputCount; ++i) { in concatenation()
108 return concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount), in concatenation()
115 uint32_t inputCount = context->getNumInputs() - 1; in concatenation() local
116 std::vector<std::vector<uint8_t>> inputs_uint8(inputCount); in concatenation()
117 for (uint32_t i = 0; i < inputCount; ++i) { in concatenation()
126 for (uint32_t i = 0; i < inputCount; ++i) { in concatenation()
135 NN_RET_CHECK(concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount), in concatenation()
DDensify.cpp145 const uint32_t inputCount = context->getNumInputs(); in densify() local
146 inputShapes.reserve(inputCount); in densify()
159 for (uint32_t i = kInputArrSeg; i < inputCount; i++) { in densify()
/packages/modules/NeuralNetworks/runtime/test/
DTestPartitioningRandom.cpp171 uint32_t inputCount() const { return builder()->inputCount(); } in inputCount() function in android::__anon661516c00111::TestModel
447 const uint32_t inputCount = stats.mBuilder->inputCount(); in operator <<() local
450 << ", operandCount = " << operandCount << ", inputCount = " << inputCount << " (" in operator <<()
451 << (double(inputCount) / operandCount) << ")" in operator <<()
1260 std::vector<float> goldenInputs(problemSize * problemSize * model.inputCount()); in TEST_P()
1285 std::vector<InputOutputDescriptor> ioDescriptors(model.inputCount() + model.outputCount()); in TEST_P()
1287 ioDescriptors[i].mKind = (i < model.inputCount() ? InputOutputDescriptor::INPUT in TEST_P()
1355 CHECK(inputIndex == model.inputCount()); in TEST_P()
DTestPartitioning.cpp1266 ASSERT_LT(model->inputCount(), kPseudoDefiningOperationRange); in buildDefinitionMap()
1267 for (uint32_t i = 0, e = model->inputCount(); i < e; i++) { in buildDefinitionMap()
1381 modelA->inputCount() != modelB->inputCount() || in compare()
/packages/apps/TV/src/com/android/tv/
DTvApplication.java398 int inputCount = 0; in handleTvInputKey() local
403 ++inputCount; in handleTvInputKey()
407 ++inputCount; in handleTvInputKey()
410 if (inputCount < 2) { in handleTvInputKey()
/packages/modules/NeuralNetworks/runtime/include/
DNeuralNetworks.h1137 ANeuralNetworksOperationType type, uint32_t inputCount,
1165 int ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount,
/packages/modules/NeuralNetworks/shim_and_sl/public/
DNeuralNetworksSupportLibraryImpl.h1108 ANeuralNetworksOperationType type, uint32_t inputCount,
1175 uint32_t inputCount,
/packages/modules/NeuralNetworks/common/include/
DLegacyUtils.h238 int validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount,
/packages/modules/NeuralNetworks/extensions/
DREADME.md153 CHECK_EQ(ANeuralNetworksModel_addOperation(model, type, inputCount, inputs, outputCount, outputs),