Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp139 void markProcessed(uint32_t operationIndex, OperationReadyCallback cb);
151 for (uint32_t operationIndex = 0; operationIndex < operations.size(); operationIndex++) { in OperandTracker() local
152 const Operation& operation = operations[operationIndex]; in OperandTracker()
159 mOperandToOperations.emplace(operandIndex, operationIndex); in OperandTracker()
163 cb(operationIndex); in OperandTracker()
165 mUnknownInputCount[operationIndex] = count; in OperandTracker()
169 void OperandTracker::markProcessed(uint32_t operationIndex, OperationReadyCallback cb) { in markProcessed() argument
171 const Operation& operation = mModel->getOperations()[operationIndex]; in markProcessed()
304 int ExecutionStep::addOperation(int operationIndex) { in addOperation() argument
305 const Operation& operation = getSourceModel()->getOperation(operationIndex); in addOperation()
[all …]
DModelBuilder.h153 uint32_t operationIndex) const;
154 bool supportedByControlFlowInterpreter(uint32_t operationIndex) const;
158 bool isControlFlowOperationWithOperandOfUnknownSize(uint32_t operationIndex) const;
DModelBuilder.cpp400 uint32_t operationIndex = operationCount(); in addOperation() local
401 if (operationIndex >= MAX_NUMBER_OF_OPERATIONS) { in addOperation()
816 for (uint32_t operationIndex = 0; operationIndex < operationCount(); operationIndex++) { in sortIntoRunOrder() local
817 uint32_t& count = unknownInputCount[operationIndex]; in sortIntoRunOrder()
819 for (uint32_t operandIndex : mOperations[operationIndex].inputs) { in sortIntoRunOrder()
825 std::pair<uint32_t, uint32_t>(operandIndex, operationIndex)); in sortIntoRunOrder()
829 opsReadyToRun.push_back(operationIndex); in sortIntoRunOrder()
DManager.cpp210 for (size_t operationIndex = 0; operationIndex < supportedOperations.size(); operationIndex++) { in getSupportedOperations() local
211 if (!supportedOperations[operationIndex]) { in getSupportedOperations()
216 const Operation& operation = hidlModel.main.operations[operationIndex]; in getSupportedOperations()
235 supportedOperations[operationIndex] = false; in getSupportedOperations()
DExecutionPlan.h119 int addOperation(int operationIndex);
/frameworks/ml/nn/runtime/test/
DTestPartitioningRandom.cpp150 const uint32_t operationIndex = operationCount(); in addOperation() local
153 return operationIndex; in addOperation()
1014 const uint32_t operationIndex = model.addOperation(operationPattern.mOperationType, in TEST_P() local
1016 deadOperations.insert(operationIndex); in TEST_P()
1018 [&deadOperands, operationIndex](uint32_t operandIndex) { in TEST_P()
1019 deadOperands.insert(std::make_pair(operandIndex, operationIndex)); in TEST_P()
DTestPartitioning.cpp253 uint32_t operationIndex) { in lookupOperation() argument
254 const Operation& operation = getOperation(operationIndex); in lookupOperation()
283 uint32_t operationIndex) { in lookupOperation() argument
287 [&model](uint32_t offset) { return &model.operandValues[offset]; }, operationIndex); in lookupOperation()