Searched defs:deviceCount (Results 1 - 1 of 1) sorted by relevance

/frameworks/ml/nn/runtime/
H A DExecutionPlan.cpp773 // Note that deviceCount includes CPU, which has no entry in devices[].
774 const size_t deviceCount = nonCpuDeviceCount + 1; local
777 VLOG(COMPILATION) << "ModelBuilder::partitionTheWork: deviceCount = " << deviceCount
797 int status = findBestDeviceForEachOperation(preference, devices, deviceCount,
807 const bool cpu = (size_t(bestDeviceIndex) == deviceCount - 1);
818 std::vector<std::queue<uint32_t>> perDeviceQueue(deviceCount);
834 for (int i = deviceCount - 1; i >= 0; i--) {
933 const size_t deviceCount,
936 // Note that deviceCount include
930 findBestDeviceForEachOperation( uint32_t preference, const std::vector<std::shared_ptr<Device>>& devices, const size_t deviceCount, std::vector<int>* bestDeviceForOperation) const argument
[all...]

Completed in 716 milliseconds