Lines Matching refs:operationIndex
139 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()
308 mToken.update(&operationIndex, sizeof(operationIndex)); in addOperation()
1672 auto enqueueOnAppropriateDevice = [&](uint32_t operationIndex) { in partitionTheWorkInternal() argument
1673 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWorkInternal()
1674 perDeviceQueue[deviceIndex].push(operationIndex); in partitionTheWorkInternal()
1675 VLOG(COMPILATION) << "enqueueOnAppropriateDevice " << operationIndex << " onto " in partitionTheWorkInternal()
1710 uint32_t operationIndex = queue.front(); in partitionTheWorkInternal() local
1712 int n = step->addOperation(operationIndex); in partitionTheWorkInternal()
1714 LOG(ERROR) << "failed to add operation " << operationIndex << " to step"; in partitionTheWorkInternal()
1717 tracker.markProcessed(operationIndex, enqueueOnAppropriateDevice); in partitionTheWorkInternal()
1721 uint32_t operationIndex = queue.front(); in partitionTheWorkInternal() local
1723 const Operation& operation = getOperation(operationIndex); in partitionTheWorkInternal()
1843 tracker.markProcessed(operationIndex, enqueueOnAppropriateDevice); in partitionTheWorkInternal()
1861 for (size_t operationIndex = 0; operationIndex < operationCount; operationIndex++) { in getPerformance() local
1862 perf += getPerformance(preference, device, operationIndex); in getPerformance()
1868 uint32_t operationIndex) const { in getPerformance()
1873 const Operation& operation = getOperation(operationIndex); in getPerformance()
1919 bool ModelBuilder::isControlFlowOperationWithOperandOfUnknownSize(uint32_t operationIndex) const { in isControlFlowOperationWithOperandOfUnknownSize()
1930 const Operation& operation = getOperation(operationIndex); in isControlFlowOperationWithOperandOfUnknownSize()
1964 bool ModelBuilder::supportedByControlFlowInterpreter(uint32_t operationIndex) const { in supportedByControlFlowInterpreter()
1965 const Operation& operation = getOperation(operationIndex); in supportedByControlFlowInterpreter()
1968 !isControlFlowOperationWithOperandOfUnknownSize(operationIndex); in supportedByControlFlowInterpreter()
1982 bool check(size_t operationIndex) const { return mSupportsOperationByIndex[operationIndex]; } in check()
2003 for (size_t operationIndex = 0; operationIndex < operationCount; operationIndex++) { in findBestDeviceForEachOperation() local
2004 const Operation& operation = getOperation(operationIndex); in findBestDeviceForEachOperation()
2008 if (isControlFlowOperationWithOperandOfUnknownSize(operationIndex)) { in findBestDeviceForEachOperation()
2016 if (canDo[cpuDeviceIndex].check(operationIndex)) { in findBestDeviceForEachOperation()
2024 if (canDo[deviceIndex].check(operationIndex)) { in findBestDeviceForEachOperation()
2025 const float perfVal = getPerformance(preference, device, operationIndex); in findBestDeviceForEachOperation()
2047 supportedByControlFlowInterpreter(operationIndex)) { in findBestDeviceForEachOperation()
2051 (*bestDeviceForOperation)[operationIndex] = kControlFlowInterpreter; in findBestDeviceForEachOperation()
2056 (*bestDeviceForOperation)[operationIndex] = bestChoice; in findBestDeviceForEachOperation()