Home
last modified time | relevance | path

Searched refs:operationIndex (Results 1 – 7 of 7) sorted by relevance

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp218 void markProcessed(uint32_t operationIndex, OperationReadyCallback cb);
230 for (uint32_t operationIndex = 0; operationIndex < operations.size(); operationIndex++) { in OperandTracker() local
231 const Operation& operation = operations[operationIndex]; in OperandTracker()
239 std::pair<uint32_t, uint32_t>(operandIndex, operationIndex)); in OperandTracker()
243 cb(operationIndex); in OperandTracker()
245 mUnknownInputCount[operationIndex] = count; in OperandTracker()
249 void OperandTracker::markProcessed(uint32_t operationIndex, OperationReadyCallback cb) { in markProcessed() argument
251 const Operation& operation = mModel->getOperations()[operationIndex]; in markProcessed()
370 int ExecutionStep::addOperation(int operationIndex, const ModelBuilder& fromModel) { in addOperation() argument
371 const Operation& operation = fromModel.getOperation(operationIndex); in addOperation()
[all …]
DModelBuilder.cpp344 uint32_t operationIndex = operationCount(); in addOperation() local
345 if (operationIndex >= MAX_NUMBER_OF_OPERATIONS) { in addOperation()
490 for (uint32_t operationIndex = 0; operationIndex < operationCount(); operationIndex++) { in sortIntoRunOrder() local
491 uint32_t& count = unknownInputCount[operationIndex]; in sortIntoRunOrder()
493 for (uint32_t operandIndex : mOperations[operationIndex].inputs) { in sortIntoRunOrder()
499 std::pair<uint32_t, uint32_t>(operandIndex, operationIndex)); in sortIntoRunOrder()
503 opsReadyToRun.push_back(operationIndex); in sortIntoRunOrder()
DManager.cpp192 for (size_t operationIndex = 0; operationIndex < outSupportedOperations->size(); in getSupportedOperations() local
193 operationIndex++) { in getSupportedOperations()
194 if (!(*outSupportedOperations)[operationIndex]) { in getSupportedOperations()
199 const Operation &operation = hidlModel.operations[operationIndex]; in getSupportedOperations()
218 (*outSupportedOperations)[operationIndex] = false; in getSupportedOperations()
DModelBuilder.h113 uint32_t operationIndex) const;
DExecutionPlan.h55 int addOperation(int operationIndex, const ModelBuilder& fromModel);
/frameworks/ml/nn/runtime/test/
DTestPartitioningRandom.cpp153 const uint32_t operationIndex = operationCount(); in addOperation() local
156 return operationIndex; in addOperation()
961 const uint32_t operationIndex = in TEST_P() local
964 deadOperations.insert(operationIndex); in TEST_P()
966 [&deadOperands, operationIndex](uint32_t operandIndex) { in TEST_P()
967 deadOperands.insert(std::make_pair(operandIndex, operationIndex)); in TEST_P()
DTestPartitioning.cpp228 uint32_t operationIndex) { in lookupOperation() argument
229 const Operation& operation = getOperation(operationIndex); in lookupOperation()
259 uint32_t lookupOperation(const HidlModel& model, uint32_t operationIndex) { in lookupOperation() argument
268 operationIndex); in lookupOperation()