Searched refs:SetDefaultDevice (Results 1 – 5 of 5) sorted by relevance
29 inline void SetDefaultDevice(const string& device, GraphDef* graph_def) { in SetDefaultDevice() function
101 graph::SetDefaultDevice("/gpu:0", &def); in TEST_F()
105 graph::SetDefaultDevice(opts->use_gpu ? "/device:GPU:0" : "/cpu:0", &def); in ConcurrentSteps()
178 graph::SetDefaultDevice(cluster->devices[0].name(), &def); in BM_Helper()
249 graph::SetDefaultDevice(cluster->devices()[0].name(), &graph_copy); in TEST()256 graph::SetDefaultDevice(cluster->devices()[1].name(), &graph_copy); in TEST()