Searched refs:operationIndex (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ExecutionPlan.cpp | 86 void markProcessed(uint32_t operationIndex, OperationReadyCallback cb);
98 for (uint32_t operationIndex = 0; operationIndex < operations.size(); operationIndex++) { in OperandTracker() local
99 const Operation& operation = operations[operationIndex]; in OperandTracker()
107 std::pair<uint32_t, uint32_t>(operandIndex, operationIndex)); in OperandTracker()
111 cb(operationIndex); in OperandTracker()
113 mUnknownInputCount[operationIndex] = count; in OperandTracker()
117 void OperandTracker::markProcessed(uint32_t operationIndex, OperationReadyCallback cb) { in markProcessed() argument
119 const Operation& operation = mModel->getOperations()[operationIndex]; in markProcessed()
229 int ExecutionStep::addOperation(int operationIndex, const ModelBuilder& fromModel) { in addOperation() argument
230 const Operation& operation = fromModel.getOperation(operationIndex); in addOperation()
[all …]
|
| D | Manager.cpp | 96 for (size_t operationIndex = 0; operationIndex < outSupportedOperations->size(); in getSupportedOperations() local
97 operationIndex++) { in getSupportedOperations()
98 if (!(*outSupportedOperations)[operationIndex]) { in getSupportedOperations()
103 const Operation &operation = hidlModel.operations[operationIndex]; in getSupportedOperations()
122 (*outSupportedOperations)[operationIndex] = false; in getSupportedOperations()
|
| D | ModelBuilder.cpp | 217 uint32_t operationIndex = operationCount(); in addOperation() local
218 if (operationIndex >= MAX_NUMBER_OF_OPERATIONS) { in addOperation()
351 for (uint32_t operationIndex = 0; operationIndex < operationCount(); operationIndex++) { in sortIntoRunOrder() local
352 uint32_t& count = unknownInputCount[operationIndex]; in sortIntoRunOrder()
354 for (uint32_t operandIndex : mOperations[operationIndex].inputs) { in sortIntoRunOrder()
360 std::pair<uint32_t, uint32_t>(operandIndex, operationIndex)); in sortIntoRunOrder()
364 opsReadyToRun.push_back(operationIndex); in sortIntoRunOrder()
|
| D | ModelBuilder.h | 96 uint32_t operationIndex) const;
|
| D | ExecutionPlan.h | 50 int addOperation(int operationIndex, const ModelBuilder& fromModel);
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestPartitioningRandom.cpp | 155 const uint32_t operationIndex = operationCount(); in addOperation() local
158 return operationIndex; in addOperation()
815 const uint32_t operationIndex = in TEST_P() local
818 deadOperations.insert(operationIndex); in TEST_P()
820 [&deadOperands, operationIndex](uint32_t operandIndex) { in TEST_P()
821 deadOperands.insert(std::make_pair(operandIndex, operationIndex)); in TEST_P()
|
| D | TestPartitioning.cpp | 154 uint32_t operationIndex) { in lookupOperation() argument
155 const Operation& operation = getOperation(operationIndex); in lookupOperation()
180 uint32_t lookupOperation(const HidlModel& model, uint32_t operationIndex) { in lookupOperation() argument
189 operationIndex); in lookupOperation()
|