Lines Matching refs:deviceIndex
998 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWork() local
999 perDeviceQueue[deviceIndex].push(operationIndex); in partitionTheWork()
1001 << deviceIndex; in partitionTheWork()
1022 int deviceIndex = findNextDeviceToProcess(); in partitionTheWork() local
1023 VLOG(COMPILATION) << "findNextDeviceToProcess: " << deviceIndex; in partitionTheWork()
1024 if (deviceIndex < 0) { in partitionTheWork()
1029 std::shared_ptr<ExecutionStep> step = plan->createNewStep(devices[deviceIndex]); in partitionTheWork()
1030 auto& queue = perDeviceQueue[deviceIndex]; in partitionTheWork()
1561 for (size_t deviceIndex = 0; deviceIndex < deviceCount; deviceIndex++) { in findBestDeviceForEachOperation() local
1562 canDo[deviceIndex].initialize(&slicer, devices[deviceIndex]); in findBestDeviceForEachOperation()
1571 for (size_t deviceIndex = 0; deviceIndex < deviceCount; deviceIndex++) { in findBestDeviceForEachOperation() local
1572 const auto& device = devices[deviceIndex]; in findBestDeviceForEachOperation()
1573 if (canDo[deviceIndex].check(operationIndex)) { in findBestDeviceForEachOperation()
1580 bestChoice = deviceIndex; in findBestDeviceForEachOperation()