Searched refs:deviceCount (Results 1 – 4 of 4) sorted by relevance
235 int deviceCount = in.readInt();236 for (int i = 0; i < deviceCount; i++) {
205 int deviceCount = in.readInt();206 for (int i = 0; i < deviceCount; i++) {
774 const size_t deviceCount = nonCpuDeviceCount + 1; in partitionTheWork() local777 VLOG(COMPILATION) << "ModelBuilder::partitionTheWork: deviceCount = " << deviceCount in partitionTheWork()797 int status = findBestDeviceForEachOperation(preference, devices, deviceCount, in partitionTheWork()807 const bool cpu = (size_t(bestDeviceIndex) == deviceCount - 1); in partitionTheWork()818 std::vector<std::queue<uint32_t>> perDeviceQueue(deviceCount); in partitionTheWork()834 for (int i = deviceCount - 1; i >= 0; i--) { in partitionTheWork()933 const size_t deviceCount, in findBestDeviceForEachOperation() argument937 const size_t nonCpuDeviceCount = deviceCount - 1; in findBestDeviceForEachOperation()
93 const size_t deviceCount,