Searched defs:operationIndex (Results 1 - 3 of 3) sorted by relevance

/frameworks/ml/nn/runtime/
H A DModelBuilder.cpp196 uint32_t operationIndex = operationCount(); local
197 if (operationIndex >= MAX_NUMBER_OF_OPERATIONS) {
201 mOperations.resize(operationIndex + 1);
202 auto& entry = mOperations[operationIndex];
209 // TODO mOperands[i].consumers.push_back(operationIndex);
304 for (uint32_t operationIndex = 0; operationIndex < operationCount(); operationIndex++) {
305 uint32_t& count = unknownInputCount[operationIndex];
307 for (uint32_t operandIndex : mOperations[operationIndex]
[all...]
H A DExecutionPlan.cpp87 void markProcessed(uint32_t operationIndex, OperationReadyCallback cb);
99 for (uint32_t operationIndex = 0; operationIndex < operations.size(); operationIndex++) {
100 const Operation& operation = operations[operationIndex];
108 std::pair<uint32_t, uint32_t>(operandIndex, operationIndex));
112 cb(operationIndex);
114 mUnknownInputCount[operationIndex] = count;
118 void OperandTracker::markProcessed(uint32_t operationIndex, OperationReadyCallback cb) { argument
120 const Operation& operation = mModel->getOperations()[operationIndex];
222 addOperation(int operationIndex, const ModelBuilder& fromModel) argument
744 uint32_t operationIndex = queue.front(); local
[all...]
/frameworks/ml/nn/runtime/test/
H A DTestPartitioning.cpp138 uint32_t operationIndex) {
139 const Operation& operation = getOperation(operationIndex);
164 uint32_t lookupOperation(const HidlModel& model, uint32_t operationIndex) { argument
173 operationIndex); member in namespace:__anon1497
135 lookupOperation(std::function<const Operation&(uint32_t)> getOperation, std::function<const Operand&(uint32_t)> getOperand, std::function<const uint8_t*(uint32_t)> getValue, uint32_t operationIndex) argument

Completed in 118 milliseconds