Searched refs:get_canonical_device_name (Results 1 – 6 of 6) 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 …]
118 string device = get_canonical_device_name(node); in get_device()126 string VirtualPlacer::get_canonical_device_name(const NodeDef& node) const { in get_canonical_device_name() function in tensorflow::grappler::VirtualPlacer
38 string get_canonical_device_name(const NodeDef& node) const;
527 return placer_.get_canonical_device_name(*node); in DeviceName()533 return str_util::StringReplace(placer_.get_canonical_device_name(*node), ":", in SanitizedDeviceName()
426 device_name = virtual_placer_.get_canonical_device_name(*node); in AddNodePermConst()450 device_name = virtual_placer_.get_canonical_device_name(*node); in AddNodeConstScalar()574 device_name = virtual_placer_.get_canonical_device_name(*node_); in IsOnGPU()
1228 string device = virtual_placer.get_canonical_device_name(node); in TEST_F()