Home
last modified time | relevance | path

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

/aosp12/build/soong/cmd/multiproduct_kati/
H A Dmain_test.go25 inputCount int
30 inputCount: 1,
35 inputCount: 1,
40 inputCount: 4,
45 inputCount: 19,
61 inputCount: 15,
79 t.Run(fmt.Sprintf("%d/%d", tc.inputCount, tc.shardCount), func(t *testing.T) {
81 for i := 1; i <= tc.inputCount; i++ {
89 tc.inputCount, tc.shardCount, tc.want, got)
/aosp12/packages/modules/NeuralNetworks/common/
H A DLegacyUtils.cpp185 inputCount(inputCount), in OperationValidationContext()
207 uint32_t inputCount; member in android::nn::__anon648133630110::OperationValidationContext
229 return inputCount; in getNumInputs()
804 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
833 if (inputCount == 3) { in validateOperation()
844 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
873 if (inputCount == 3) { in validateOperation()
1025 if ((inputCount != 23 && inputCount != 27) || outputCount != 4) { in validateOperation()
1175 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
1225 if ((inputCount != 4 && inputCount != 3) || outputCount != 1) { in validateOperation()
[all …]
H A DLegacyHalUtils.cpp203 static inline int validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount, in validateOperation() argument
212 return validateOperation(opType, inputCount, inputIndexes, outputCount, outputIndexes, operands, in validateOperation()
/aosp12/packages/modules/NeuralNetworks/runtime/
H A DModelBuilder.cpp653 if (10 < inputCount && inputCount <= 13 && in getNumTrailingArgumentsToRemove()
661 } else if (7 < inputCount && inputCount <= 10 && in getNumTrailingArgumentsToRemove()
672 if (11 < inputCount && inputCount <= 14 && in getNumTrailingArgumentsToRemove()
680 } else if (8 < inputCount && inputCount <= 11 && in getNumTrailingArgumentsToRemove()
691 if (inputCount == 3) { in getNumTrailingArgumentsToRemove()
698 if (inputCount == 2) { in getNumTrailingArgumentsToRemove()
719 if (inputCount == 6) { in getNumTrailingArgumentsToRemove()
740 if (3 < inputCount && inputCount <= 6) { in getNumTrailingArgumentsToRemove()
752 if (inputCount == 3) { in getNumTrailingArgumentsToRemove()
759 if (inputCount == 3) { in getNumTrailingArgumentsToRemove()
[all …]
H A DModelBuilder.h54 int addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, const uint32_t* inputs,
56 int identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount,
85 uint32_t inputCount() const { return static_cast<uint32_t>(mInputIndexes.size()); } in inputCount() function
H A DCompilationBuilder.cpp258 if (index >= mModel->inputCount()) { in getPreferredMemoryAlignmentForInput()
280 if (index >= mModel->inputCount()) { in getPreferredMemoryPaddingForInput()
381 if (index >= mModel->inputCount()) { in forEachStepRoleOfInput()
H A DExecutionPlan.cpp539 const uint32_t inputCount = static_cast<uint32_t>(operation.inputs.size()); in addOperation() local
541 std::vector<uint32_t> inputs(inputCount); in addOperation()
545 return mStepModel.addOperation(static_cast<uint32_t>(operation.type), inputCount, inputs.data(), in addOperation()
780 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finishStepModel()
949 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finish()
1991 CHECK_LT(index, mainModel->inputCount()); in getInputSourceOperand()
2259 for (uint32_t i = 0, n = thenModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2268 for (uint32_t i = 0, n = elseModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2320 for (uint32_t i = 0, n = condModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2327 for (uint32_t i = 0, n = bodyModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
H A DNeuralNetworks.cpp1137 ANeuralNetworksOperationType type, uint32_t inputCount, in ANeuralNetworksModel_addOperation() argument
1146 return m->addOperation(type, inputCount, inputs, outputCount, outputs); in ANeuralNetworksModel_addOperation()
1162 int ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount, in ANeuralNetworksModel_identifyInputsAndOutputs() argument
1171 return m->identifyInputsAndOutputs(inputCount, inputs, outputCount, outputs); in ANeuralNetworksModel_identifyInputsAndOutputs()
H A DExecutionBuilder.cpp157 mInputs(mModel->inputCount()), in ExecutionBuilder()
1323 mInputs(model->inputCount()), in StepExecutor()
H A DMemory.cpp339 if (index >= model->inputCount()) { in addRole()
/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DConcatenation.cpp97 uint32_t inputCount = context->getNumInputs() - 1; in concatenation() local
100 for (uint32_t i = 0; i < inputCount; ++i) { in concatenation()
113 uint32_t inputCount = context->getNumInputs() - 1; variable
114 std::vector<std::vector<uint8_t>> inputs_uint8(inputCount);
115 for (int i = 0; i < inputCount; ++i) {
124 for (uint32_t i = 0; i < inputCount; ++i) {
145 uint32_t inputCount = context->getNumInputs(); in validate() local
146 NN_RET_CHECK_GE(inputCount, 2); in validate()
159 std::vector<OperandType> inExpectedTypes(inputCount - 1, inputType); in validate()
163 for (uint32_t i = 0; i < inputCount - 1; ++i) { in validate()
[all …]
H A DPooling.cpp298 auto inputCount = context->getNumInputs(); in validate() local
299 NN_RET_CHECK(inputCount == 11 || inputCount == 10 || inputCount == 8 || inputCount == 7); in validate()
344 if (inputCount >= 10) { in validate()
349 if (inputCount == 11 || inputCount == 8) { in validate()
H A DConv2D.cpp547 auto inputCount = context->getNumInputs(); in validate() local
598 if (inputCount >= 8) { in validate()
599 if (context->getInputType(7) == OperandType::INT32 && inputCount >= 10) { in validate()
606 if (inputCount >= 8 + inputOffset) { in validate()
610 NN_RET_CHECK_NE(inputCount, 9 + inputOffset) in validate()
613 if (inputCount == 10 + inputOffset) { in validate()
H A DTransposeConv2D.cpp442 const uint32_t inputCount = context->getNumInputs(); in validate() local
443 NN_RET_CHECK(inputCount == kNumInputs1 || inputCount == kNumInputs2); in validate()
473 if (inputCount == 11) { in validate()
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestPartitioningRandom.cpp174 uint32_t inputCount() const { return builder()->inputCount(); } in inputCount() function in android::__anonc614c5800110::TestModel
450 const uint32_t inputCount = stats.mBuilder->inputCount(); in operator <<() local
453 << ", operandCount = " << operandCount << ", inputCount = " << inputCount << " (" in operator <<()
454 << (double(inputCount) / operandCount) << ")" in operator <<()
1276 std::vector<float> goldenInputs(problemSize * problemSize * model.inputCount()); in TEST_P()
1301 std::vector<InputOutputDescriptor> ioDescriptors(model.inputCount() + model.outputCount()); in TEST_P()
1303 ioDescriptors[i].mKind = (i < model.inputCount() ? InputOutputDescriptor::INPUT in TEST_P()
1371 CHECK(inputIndex == model.inputCount()); in TEST_P()
H A DTestPartitioning.cpp1265 ASSERT_LT(model->inputCount(), kPseudoDefiningOperationRange); in buildDefinitionMap()
1266 for (uint32_t i = 0, e = model->inputCount(); i < e; i++) { in buildDefinitionMap()
1380 modelA->inputCount() != modelB->inputCount() || in compare()
/aosp12/packages/apps/TV/src/com/android/tv/
H A DTvApplication.java398 int inputCount = 0; in handleTvInputKey() local
403 ++inputCount; in handleTvInputKey()
407 ++inputCount; in handleTvInputKey()
410 if (inputCount < 2) { in handleTvInputKey()
/aosp12/packages/modules/NeuralNetworks/runtime/include/
H A DNeuralNetworks.h1132 ANeuralNetworksOperationType type, uint32_t inputCount,
1160 int ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount,
/aosp12/packages/modules/NeuralNetworks/shim_and_sl/public/
H A DNeuralNetworksSupportLibraryImpl.h784 ANeuralNetworksOperationType type, uint32_t inputCount,
851 uint32_t inputCount,
/aosp12/packages/modules/NeuralNetworks/common/include/
H A DLegacyUtils.h260 int validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount,
/aosp12/packages/modules/NeuralNetworks/extensions/
H A DREADME.md153 CHECK_EQ(ANeuralNetworksModel_addOperation(model, type, inputCount, inputs, outputCount, outputs),