Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp771 const size_t nonCpuDeviceCount = devices.size(); in partitionTheWork() local
774 const size_t deviceCount = nonCpuDeviceCount + 1; in partitionTheWork()
781 if (nonCpuDeviceCount == 0 || operationCount == 0) { in partitionTheWork()
853 static_cast<size_t>(deviceIndex) < nonCpuDeviceCount in partitionTheWork()
937 const size_t nonCpuDeviceCount = deviceCount - 1; in findBestDeviceForEachOperation() local
939 std::vector<CanDo> canDo(nonCpuDeviceCount); in findBestDeviceForEachOperation()
940 for (size_t deviceIndex = 0; deviceIndex < nonCpuDeviceCount; deviceIndex++) { in findBestDeviceForEachOperation()
950 for (size_t deviceIndex = 0; deviceIndex < nonCpuDeviceCount; deviceIndex++) { in findBestDeviceForEachOperation()
983 bestChoice = nonCpuDeviceCount; // The ID of the CPU. in findBestDeviceForEachOperation()