| /frameworks/ml/nn/common/ |
| D | Utils.cpp | 197 const uint32_t* inputIndexes, uint32_t outputCount, in OperationValidationContext() argument
203 outputCount(outputCount), in OperationValidationContext()
227 uint32_t outputCount; member in android::nn::__anon8d54713c0111::OperationValidationContext
247 CHECK(index < static_cast<uint32_t>(outputCount)); in getOutputOperand()
256 return outputCount; in getNumOutputs()
730 static bool validateIfOperation(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, in validateIfOperation() argument
736 NN_RET_CHECK_GE(outputCount, 1u) << "ANEURALNETWORKS_IF must have at least 1 output"; in validateIfOperation()
743 NN_RET_CHECK_EQ(outputCount, branchModelOutputCount); in validateIfOperation()
775 uint32_t outputCount, const uint32_t* outputs, in validateWhileOperation() argument
789 NN_RET_CHECK_GE(outputCount, 1u) << "ANEURALNETWORKS_WHILE must have at least 1 output"; in validateWhileOperation()
[all …]
|
| /frameworks/ml/nn/runtime/ |
| D | ModelBuilder.h | 55 uint32_t outputCount, const uint32_t* outputs);
56 int identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount,
86 uint32_t outputCount() const { return static_cast<uint32_t>(mOutputIndexes.size()); } in outputCount() function
|
| D | ModelBuilder.cpp | 350 const uint32_t* inputs, uint32_t outputCount, in addOperation() argument
383 return getReferencedModel(modelOperand)->outputCount(); in addOperation()
392 type, inputCount, inputs, outputCount, outputs, mOperands, HalVersion::LATEST, in addOperation()
409 .outputs = hidl_vec<uint32_t>(outputs, outputs + outputCount), in addOperation()
421 uint32_t outputCount, const uint32_t* outputs) { in identifyInputsAndOutputs() argument
431 n = validateOperandList(outputCount, outputs, operandCount(), in identifyInputsAndOutputs()
467 !setArguments(&mOutputIndexes, outputCount, outputs, OperandLifeTime::SUBGRAPH_OUTPUT)) { in identifyInputsAndOutputs()
|
| D | CompilationBuilder.cpp | 241 if (index >= mModel->outputCount()) { in forEachStepRoleOfOutput()
|
| D | ExecutionPlan.cpp | 329 const uint32_t outputCount = static_cast<uint32_t>(operation.outputs.size()); in addOperation() local
331 std::vector<uint32_t> outputs(outputCount); in addOperation()
335 outputCount, outputs.data()); in addOperation()
510 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finishStepModel()
670 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finish()
1768 for (uint32_t i = 0, n = thenModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
1777 for (uint32_t i = 0, n = elseModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
1836 for (uint32_t i = 0, n = bodyModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
|
| D | NeuralNetworks.cpp | 1113 const uint32_t* inputs, uint32_t outputCount, in ANeuralNetworksModel_addOperation() argument
1121 return m->addOperation(type, inputCount, inputs, outputCount, outputs); in ANeuralNetworksModel_addOperation()
1138 const uint32_t* inputs, uint32_t outputCount, in ANeuralNetworksModel_identifyInputsAndOutputs() argument
1146 return m->identifyInputsAndOutputs(inputCount, inputs, outputCount, outputs); in ANeuralNetworksModel_identifyInputsAndOutputs()
|
| D | ExecutionBuilder.cpp | 96 mOutputs(mModel->outputCount()) { in ExecutionBuilder()
987 mOutputs(model->outputCount()) { in StepExecutor()
|
| D | Memory.cpp | 360 if (index >= model->outputCount()) { in addRole()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestPartitioningRandom.cpp | 159 uint32_t outputCount() const { return builder()->outputCount(); } in outputCount() function in android::__anonbe695bce0111::TestModel
435 const uint32_t outputCount = stats.mBuilder->outputCount(); in operator <<() local
439 << ", outputCount = " << outputCount << " (" << (double(outputCount) / operandCount) in operator <<()
1046 for (uint32_t outputIdx = 0, outputCount = outputs.size(); outputIdx < outputCount; in TEST_P() local
1271 std::vector<InputOutputDescriptor> ioDescriptors(model.inputCount() + model.outputCount()); in TEST_P()
1342 CHECK(outputIndex == model.outputCount()); in TEST_P()
1351 std::vector<float> nonPartitionedOutputs(problemSize * problemSize * model.outputCount()); in TEST_P()
|
| D | TestPartitioning.cpp | 1248 modelA->outputCount() != modelB->outputCount()) { in compare()
1269 for (uint32_t i = 0, e = modelA->outputCount(); i < e; i++) { in compare()
|
| /frameworks/ml/nn/runtime/include/ |
| D | NeuralNetworks.h | 7376 const uint32_t* inputs, uint32_t outputCount,
7404 const uint32_t* inputs, uint32_t outputCount,
|
| /frameworks/base/core/jni/android/opengl/ |
| D | util.cpp | 471 int outputCount; in util_frustumCullSpheres() local
493 outputCount = 0; in util_frustumCullSpheres()
496 if (outputCount < resultsCapacity) { in util_frustumCullSpheres()
499 outputCount++; in util_frustumCullSpheres()
503 return outputCount; in util_frustumCullSpheres()
|
| /frameworks/ml/nn/extensions/ |
| D | README.md | 153 CHECK_EQ(ANeuralNetworksModel_addOperation(model, type, inputCount, inputs, outputCount, outputs),
|
| /frameworks/ml/nn/common/include/ |
| D | Utils.h | 407 const uint32_t* inputIndexes, uint32_t outputCount,
|
| /frameworks/ml/nn/tools/api/ |
| D | NeuralNetworks.t | 1785 * @param outputCount The number of entries in the outputs array.
1802 const uint32_t* inputs, uint32_t outputCount,
1815 * @param outputCount The number of entries in the outputs array.
1830 const uint32_t* inputs, uint32_t outputCount,
|