Searched defs:inputCount (Results 1 – 10 of 10) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Concatenation.cpp | 95 uint32_t inputCount = context->getNumInputs() - 1; in concatenation() local
111 uint32_t inputCount = context->getNumInputs() - 1; in concatenation() local
142 uint32_t inputCount = context->getNumInputs(); in validate() local
|
| D | Pooling.cpp | 295 auto inputCount = context->getNumInputs(); in validate() local
|
| D | TransposeConv2D.cpp | 440 const uint32_t inputCount = context->getNumInputs(); in validate() local
|
| D | Conv2D.cpp | 545 auto inputCount = context->getNumInputs(); in validate() local
|
| /frameworks/ml/nn/runtime/ |
| D | ModelBuilder.cpp | 349 int ModelBuilder::addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, in addOperation()
420 int ModelBuilder::identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, in identifyInputsAndOutputs()
566 const uint32_t inputCount = operation.inputs.size(); in removeTrailingArgumentsWithDefaultValues() local
618 const uint32_t inputCount = operation.inputs.size(); in countMatchingTrailingArguments() local
631 const uint32_t inputCount = operation.inputs.size(); in getNumTrailingArgumentsToRemove() local
|
| D | ModelBuilder.h | 85 uint32_t inputCount() const { return static_cast<uint32_t>(mInputIndexes.size()); } in inputCount() function
|
| D | NeuralNetworks.cpp | 1112 ANeuralNetworksOperationType type, uint32_t inputCount, in ANeuralNetworksModel_addOperation()
1137 int ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount, in ANeuralNetworksModel_identifyInputsAndOutputs()
|
| D | ExecutionPlan.cpp | 328 const uint32_t inputCount = static_cast<uint32_t>(operation.inputs.size()); in addOperation() local
|
| /frameworks/ml/nn/common/ |
| D | Utils.cpp | 196 OperationValidationContext(const char* operationName, uint32_t inputCount, in OperationValidationContext()
225 uint32_t inputCount; member in android::nn::__anon8d54713c0111::OperationValidationContext
730 static bool validateIfOperation(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, in validateIfOperation()
774 static bool validateWhileOperation(uint32_t inputCount, const uint32_t* inputs, in validateWhileOperation()
847 static inline int validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount, in validateOperation()
861 int validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount, in validateOperation()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestPartitioningRandom.cpp | 158 uint32_t inputCount() const { return builder()->inputCount(); } in inputCount() function in android::__anonbe695bce0111::TestModel
434 const uint32_t inputCount = stats.mBuilder->inputCount(); in operator <<() local
|