Home
last modified time | relevance | path

Searched defs:inputCount (Results 1 – 11 of 11) sorted by relevance

/frameworks/ml/nn/common/operations/
DConcatenation.cpp84 uint32_t inputCount = context->getNumInputs() - 1; in concatenation() local
101 uint32_t inputCount = context->getNumInputs(); in validate() local
DPooling.cpp252 auto inputCount = context->getNumInputs(); in validate() local
DTransposeConv2D.cpp429 auto inputCount = context->getNumInputs(); in validate() local
DConv2D.cpp402 auto inputCount = context->getNumInputs(); in validate() local
/frameworks/ml/nn/runtime/
DModelBuilder.cpp319 int ModelBuilder::addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, in addOperation()
364 int ModelBuilder::identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, in identifyInputsAndOutputs()
DModelBuilder.h81 uint32_t inputCount() const { return static_cast<uint32_t>(mInputIndexes.size()); } in inputCount() function
DNeuralNetworks.cpp928 ANeuralNetworksOperationType type, uint32_t inputCount, in ANeuralNetworksModel_addOperation()
953 int ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount, in ANeuralNetworksModel_identifyInputsAndOutputs()
DExecutionPlan.cpp384 const uint32_t inputCount = static_cast<uint32_t>(operation.inputs.size()); in addOperation() local
/frameworks/rs/rsov/compiler/
DWrapper.cpp388 const uint32_t *inputCount = metadata.getExportForEachInputCountList(); in AddWrappers() local
/frameworks/ml/nn/common/
DUtils.cpp122 OperationValidationContext(uint32_t inputCount, const uint32_t* inputIndexes, in OperationValidationContext()
147 uint32_t inputCount; member in android::nn::__anon6bc5d5a50111::OperationValidationContext
543 int validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount, in validateOperation()
/frameworks/ml/nn/runtime/test/
DTestPartitioningRandom.cpp163 uint32_t inputCount() const { in inputCount() function in android::__anon65ec02170111::TestModel
449 const uint32_t inputCount = stats.mBuilder->inputCount(); in operator <<() local