Lines Matching refs:deviceIndex
2153 deviceCount](int deviceIndex) -> std::string { in partitionTheWorkInternal() argument
2154 if (deviceIndex == kControlFlowInterpreter) { in partitionTheWorkInternal()
2156 } else if (deviceIndex < 0 || size_t(deviceIndex) >= deviceCount) { in partitionTheWorkInternal()
2159 return devices.at(deviceIndex)->getName(); in partitionTheWorkInternal()
2165 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWorkInternal() local
2166 perDeviceQueue[deviceIndex].push(operationIndex); in partitionTheWorkInternal()
2168 << deviceIndex << " (" << deviceName(deviceIndex) << ")"; in partitionTheWorkInternal()
2192 int deviceIndex = findNextDeviceToProcess(); in partitionTheWorkInternal() local
2193 VLOG(COMPILATION) << "findNextDeviceToProcess: " << deviceIndex << " (" in partitionTheWorkInternal()
2194 << deviceName(deviceIndex) << ")"; in partitionTheWorkInternal()
2195 if (deviceIndex < 0) { in partitionTheWorkInternal()
2200 auto& queue = perDeviceQueue[deviceIndex]; in partitionTheWorkInternal()
2201 if (deviceIndex != kControlFlowInterpreter) { in partitionTheWorkInternal()
2203 plan->createNewExecutionStep(sourceModelIndex, devices[deviceIndex]); in partitionTheWorkInternal()
2492 for (size_t deviceIndex = 0; deviceIndex < deviceCount; deviceIndex++) { in findBestDeviceForEachOperation() local
2493 canDo[deviceIndex].initialize(metaModel, devices[deviceIndex]); in findBestDeviceForEachOperation()
2518 for (size_t deviceIndex = 0; deviceIndex < deviceCount; deviceIndex++) { in findBestDeviceForEachOperation() local
2519 const auto& device = devices[deviceIndex]; in findBestDeviceForEachOperation()
2520 if (canDo[deviceIndex].check(operationIndex)) { in findBestDeviceForEachOperation()
2527 bestChoice = deviceIndex; in findBestDeviceForEachOperation()