Home
last modified time | relevance | path

Searched refs:deviceIndex (Results 1 – 1 of 1) sorted by relevance

/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp2206 deviceCount](int deviceIndex) -> std::string { in partitionTheWorkInternal() argument
2207 if (deviceIndex == kControlFlowInterpreter) { in partitionTheWorkInternal()
2209 } else if (deviceIndex < 0 || size_t(deviceIndex) >= deviceCount) { in partitionTheWorkInternal()
2212 return devices.at(deviceIndex)->getName(); in partitionTheWorkInternal()
2218 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWorkInternal() local
2219 perDeviceQueue[deviceIndex].push(operationIndex); in partitionTheWorkInternal()
2221 << deviceIndex << " (" << deviceName(deviceIndex) << ")"; in partitionTheWorkInternal()
2245 int deviceIndex = findNextDeviceToProcess(); in partitionTheWorkInternal() local
2246 VLOG(COMPILATION) << "findNextDeviceToProcess: " << deviceIndex << " (" in partitionTheWorkInternal()
2247 << deviceName(deviceIndex) << ")"; in partitionTheWorkInternal()
[all …]