Searched refs:inputCount (Results 1 - 8 of 8) sorted by relevance
/frameworks/ml/nn/runtime/ |
H A D | ModelBuilder.h | 46 int addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, const uint32_t* inputs, 48 int identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, 66 uint32_t inputCount() const { return static_cast<uint32_t>(mInputIndexes.size()); } function in class:android::nn::ModelBuilder
|
H A D | ModelBuilder.cpp | 174 int ModelBuilder::addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, argument 185 int n = validateOperandList(inputCount, inputs, operandCount(), 205 setFromIntList(&entry.inputs, inputCount, inputs); 214 int ModelBuilder::identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, argument 220 int n = validateOperandList(inputCount, inputs, operandCount(), 259 if (!setArguments(&mInputIndexes, inputCount, inputs, OperandLifeTime::MODEL_INPUT) ||
|
H A D | NeuralNetworks.cpp | 303 ANeuralNetworksOperationType type, uint32_t inputCount, 311 return m->addOperation(type, inputCount, inputs, outputCount, outputs); 314 int ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount, argument 322 return m->identifyInputsAndOutputs(inputCount, inputs, outputCount, outputs); 302 ANeuralNetworksModel_addOperation(ANeuralNetworksModel* model, ANeuralNetworksOperationType type, uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, const uint32_t* outputs) argument
|
H A D | ExecutionPlan.cpp | 233 const uint32_t inputCount = static_cast<uint32_t>(operation.inputs.size()); local 235 std::vector<uint32_t> inputs(inputCount); 258 return mSubModel->addOperation(static_cast<uint32_t>(operation.type), inputCount, inputs.data(), 311 fromModel->inputCount(),
|
H A D | ExecutionBuilder.cpp | 98 mInputs(mModel->inputCount()), 429 mInputs(model->inputCount()), mOutputs(model->outputCount()) {}
|
/frameworks/ml/nn/runtime/include/ |
H A D | NeuralNetworks.h | 1589 * @param inputCount The number of entries in the inputs array. 1605 ANeuralNetworksOperationType type, uint32_t inputCount, 1616 * @param inputCount The number of entries in the inputs array. 1630 int ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount,
|
/frameworks/rs/rsov/compiler/ |
H A D | Wrapper.cpp | 388 const uint32_t *inputCount = metadata.getExportForEachInputCountList(); local 392 AddWrapper(kernelName[i], kernelSigature[i], inputCount[i], b, m);
|
/frameworks/ml/nn/runtime/test/ |
H A D | TestPartitioning.cpp | 400 ASSERT_LT(model->inputCount(), kPseudoDefiningOperationRange); 401 for (uint32_t i = 0, e = model->inputCount(); i < e; i++) { 508 modelA->inputCount() != modelB->inputCount() ||
|
Completed in 278 milliseconds