Lines Matching refs:operationIndex
136 void markProcessed(uint32_t operationIndex, OperationReadyCallback cb);
148 for (uint32_t operationIndex = 0; operationIndex < operations.size(); operationIndex++) { in OperandTracker() local
149 const Operation& operation = operations[operationIndex]; in OperandTracker()
156 mOperandToOperations.emplace(operandIndex, operationIndex); in OperandTracker()
160 cb(operationIndex); in OperandTracker()
162 mUnknownInputCount[operationIndex] = count; in OperandTracker()
166 void OperandTracker::markProcessed(uint32_t operationIndex, OperationReadyCallback cb) { in markProcessed() argument
168 const Operation& operation = mModel->getOperations()[operationIndex]; in markProcessed()
515 int ExecutionStep::addOperation(int operationIndex) { in addOperation() argument
516 const Operation& operation = getSourceModel()->getOperation(operationIndex); in addOperation()
519 mToken.update(&operationIndex, sizeof(operationIndex)); in addOperation()
2164 auto enqueueOnAppropriateDevice = [&](uint32_t operationIndex) { in partitionTheWorkInternal() argument
2165 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWorkInternal()
2166 perDeviceQueue[deviceIndex].push(operationIndex); in partitionTheWorkInternal()
2167 VLOG(COMPILATION) << "enqueueOnAppropriateDevice " << operationIndex << " onto " in partitionTheWorkInternal()
2205 uint32_t operationIndex = queue.front(); in partitionTheWorkInternal() local
2207 int n = step->addOperation(operationIndex); in partitionTheWorkInternal()
2209 LOG(ERROR) << "failed to add operation " << operationIndex << " to step"; in partitionTheWorkInternal()
2212 tracker.markProcessed(operationIndex, enqueueOnAppropriateDevice); in partitionTheWorkInternal()
2216 uint32_t operationIndex = queue.front(); in partitionTheWorkInternal() local
2218 const Operation& operation = getOperation(operationIndex); in partitionTheWorkInternal()
2338 tracker.markProcessed(operationIndex, enqueueOnAppropriateDevice); in partitionTheWorkInternal()
2356 for (size_t operationIndex = 0; operationIndex < operationCount; operationIndex++) { in getPerformance() local
2357 perf += getPerformance(preference, device, operationIndex); in getPerformance()
2363 uint32_t operationIndex) const { in getPerformance()
2368 const Operation& operation = getOperation(operationIndex); in getPerformance()
2414 bool ModelBuilder::isControlFlowOperationWithOperandOfUnknownSize(uint32_t operationIndex) const { in isControlFlowOperationWithOperandOfUnknownSize()
2425 const Operation& operation = getOperation(operationIndex); in isControlFlowOperationWithOperandOfUnknownSize()
2459 bool ModelBuilder::supportedByControlFlowInterpreter(uint32_t operationIndex) const { in supportedByControlFlowInterpreter()
2460 const Operation& operation = getOperation(operationIndex); in supportedByControlFlowInterpreter()
2463 !isControlFlowOperationWithOperandOfUnknownSize(operationIndex); in supportedByControlFlowInterpreter()
2477 bool check(size_t operationIndex) const { return mSupportsOperationByIndex[operationIndex]; } in check()
2498 for (size_t operationIndex = 0; operationIndex < operationCount; operationIndex++) { in findBestDeviceForEachOperation() local
2499 const Operation& operation = getOperation(operationIndex); in findBestDeviceForEachOperation()
2503 if (isControlFlowOperationWithOperandOfUnknownSize(operationIndex)) { in findBestDeviceForEachOperation()
2511 if (canDo[cpuDeviceIndex].check(operationIndex)) { in findBestDeviceForEachOperation()
2520 if (canDo[deviceIndex].check(operationIndex)) { in findBestDeviceForEachOperation()
2521 const float perfVal = getPerformance(preference, device, operationIndex); in findBestDeviceForEachOperation()
2538 << operation.type << ":" << operationIndex; in findBestDeviceForEachOperation()
2547 supportedByControlFlowInterpreter(operationIndex)) { in findBestDeviceForEachOperation()
2551 (*bestDeviceForOperation)[operationIndex] = kControlFlowInterpreter; in findBestDeviceForEachOperation()
2553 << ":" << operationIndex << ") = -1 (NNAPI)"; in findBestDeviceForEachOperation()
2555 (*bestDeviceForOperation)[operationIndex] = bestChoice; in findBestDeviceForEachOperation()
2557 << ":" << operationIndex << ") = " << bestChoice << " (" in findBestDeviceForEachOperation()