Lines Matching refs:deviceIndex
822 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWork() local
823 perDeviceQueue[deviceIndex].push(operationIndex); in partitionTheWork()
825 << deviceIndex; in partitionTheWork()
846 int deviceIndex = findNextDeviceToProcess(); in partitionTheWork() local
847 VLOG(COMPILATION) << "findNextDeviceToProcess: " << deviceIndex; in partitionTheWork()
848 if (deviceIndex < 0) { in partitionTheWork()
853 static_cast<size_t>(deviceIndex) < nonCpuDeviceCount in partitionTheWork()
854 ? devices[deviceIndex] : nullptr; in partitionTheWork()
858 auto& queue = perDeviceQueue[deviceIndex]; in partitionTheWork()
940 for (size_t deviceIndex = 0; deviceIndex < nonCpuDeviceCount; deviceIndex++) { in findBestDeviceForEachOperation() local
941 canDo[deviceIndex].initialize(this, devices[deviceIndex]); in findBestDeviceForEachOperation()
950 for (size_t deviceIndex = 0; deviceIndex < nonCpuDeviceCount; deviceIndex++) { in findBestDeviceForEachOperation() local
951 const auto& device = devices[deviceIndex]; in findBestDeviceForEachOperation()
952 if (canDo[deviceIndex].check(operationIndex)) { in findBestDeviceForEachOperation()
958 bestChoice = deviceIndex; in findBestDeviceForEachOperation()