Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp2153 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()
[all …]