Searched refs:inputCount (Results 1 - 8 of 8) sorted by relevance

/frameworks/ml/nn/runtime/
H A DModelBuilder.h46 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 DModelBuilder.cpp174 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 DNeuralNetworks.cpp303 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 DExecutionPlan.cpp233 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 DExecutionBuilder.cpp98 mInputs(mModel->inputCount()),
429 mInputs(model->inputCount()), mOutputs(model->outputCount()) {}
/frameworks/ml/nn/runtime/include/
H A DNeuralNetworks.h1589 * @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 DWrapper.cpp388 const uint32_t *inputCount = metadata.getExportForEachInputCountList(); local
392 AddWrapper(kernelName[i], kernelSigature[i], inputCount[i], b, m);
/frameworks/ml/nn/runtime/test/
H A DTestPartitioning.cpp400 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