Searched refs:get_canonical_device_name (Results 1 – 8 of 8) sorted by relevance
43 placer.get_canonical_device_name(node)); in TEST()48 placer.get_canonical_device_name(node)); in TEST()53 placer.get_canonical_device_name(node)); in TEST()72 EXPECT_EQ("/GPU:0", placer.get_canonical_device_name(node)); in TEST()76 EXPECT_EQ("/CPU:0", placer.get_canonical_device_name(node)); in TEST()80 EXPECT_EQ("/GPU:0", placer.get_canonical_device_name(node)); in TEST()103 placer.get_canonical_device_name(node)); in TEST()108 placer.get_canonical_device_name(node)); in TEST()132 placer.get_canonical_device_name(node)); in TEST()136 placer.get_canonical_device_name(node)); in TEST()[all …]
116 string device = get_canonical_device_name(node); in get_device()124 string VirtualPlacer::get_canonical_device_name(const NodeDef& node) const { in get_canonical_device_name() function in tensorflow::grappler::VirtualPlacer
41 string get_canonical_device_name(const NodeDef& node) const;
638 return placer_->get_canonical_device_name(*node); in DeviceName()644 return absl::StrReplaceAll(placer_->get_canonical_device_name(*node), in SanitizedDeviceName()
434 device_name = virtual_placer_.get_canonical_device_name(*node); in AddNodePermConst()458 device_name = virtual_placer_.get_canonical_device_name(*node); in AddNodeConstScalar()586 device_name = virtual_placer_.get_canonical_device_name(*node_); in IsOnGPU()
1043 string device_name = virtual_placer_.get_canonical_device_name(node); in NodeHasF16KernelForTypeAttr()1122 device_name = virtual_placer_.get_canonical_device_name(node); in IsOnDevice()
1248 string device = virtual_placer.get_canonical_device_name(node); in TEST_F()
1956 ? virtual_placer->get_canonical_device_name(node) in GetDeviceName()