Searched refs:device0_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | partitioning_utils_test.cc | 49 device0_ = devices[0].get(); in SetUp()
61 s = s.WithDevice(device0_->name()); in SwapGraph()
72 Placer placer(graph, "", &device_set_, device0_); in SwapGraph()
88 Placer placer(graph, "", &device_set_, device0_); in TwoDeviceSwapGraph()
108 Placer placer(subgraph, "", &device_set_, device0_); in SubGraph()
113 Device* device0_ = nullptr; // Not owned. (Owned by device_mgr_.) member in tensorflow::__anone51cedca0111::PartitioningUtilsTest
|
| D | process_function_library_runtime_test.cc | 115 "/job:a/replica:0/task:0/device:CPU:0", &device0_)); in ProcessFunctionLibraryRuntimeTest()
321 Device* device0_ = nullptr; // Not owned. (Owned by device_mgr_.) member in tensorflow::__anon96a4e55d0111::ProcessFunctionLibraryRuntimeTest
353 EXPECT_EQ(flr->device(), device0_); in TEST_F()
356 EXPECT_EQ(flr->device(), device0_); in TEST_F()
359 EXPECT_EQ(flr->device(), device0_); in TEST_F()
848 ResourceMgr* mgr0 = device0_->resource_manager(); in TEST_F()
854 GetResourceHandle("var", mgr0->default_container(), device0_->name()); in TEST_F()
861 CompositeDevice::MakeDevice({device0_->name(), device1_->name()}, in TEST_F()
1197 CompositeDevice::MakeDevice({device0_->name(), device1_->name()}, in TEST_F()
|