Fixed a typo
PiperOrigin-RevId: 159038125
This commit is contained in:
parent
bea7255832
commit
e16d717e0b
@ -40,11 +40,11 @@ VirtualPlacer::VirtualPlacer(const Cluster* cluster) {
|
||||
for (const auto& device : devices_) {
|
||||
if (str_util::Lowercase(device.first).find("gpu") != string::npos) {
|
||||
default_device_ = device.first;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const DeviceProperties& VirtualPlacer::get_device(const NodeDef& node) const {
|
||||
string device = get_canonical_device_name(node);
|
||||
|
Loading…
Reference in New Issue
Block a user