Searched refs:deviceIter (Results 1 – 1 of 1) sorted by relevance
55 const auto deviceIter = deviceMap.find(c10::kCPU); in getDevicesForTensors() local56 if (deviceIter == deviceMap.end()) { in getDevicesForTensors()59 devices.emplace_back(deviceIter->second); in getDevicesForTensors()63 const auto deviceIter = deviceMap.find(t.device()); in getDevicesForTensors() local65 deviceIter != deviceMap.end(), in getDevicesForTensors()70 devices.push_back(deviceIter->second); in getDevicesForTensors()