Searched refs:first_device (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_testlib.cc | 191 const char* first_device = TFE_TensorHandleDeviceName(first, status); in Multiply() local
193 TFE_OpSetDevice(op.get(), first_device, status); in Multiply()
204 void BasicTestsForTwoDevices(TFE_Context* context, const char* first_device, in BasicTestsForTwoDevices() argument
210 std::array<const char*, 2> underlying_devices{first_device, second_device}; in BasicTestsForTwoDevices()
247 std::string first_device = in BasicTestsForTwoDevices() local
249 ASSERT_EQ(underlying_devices[0], first_device); in BasicTestsForTwoDevices()
277 std::string first_device = in BasicTestsForTwoDevices() local
279 ASSERT_EQ(underlying_devices[0], first_device); in BasicTestsForTwoDevices()
|
| D | parallel_device_test.cc | 152 std::string first_device = in TEST() local
154 ASSERT_EQ(underlying_devices[0], first_device); in TEST()
285 std::string first_device = TFE_TensorHandleBackingDeviceName( in TEST() local
287 ASSERT_EQ(second_underlying_devices[0], first_device); in TEST()
299 first_device = TFE_TensorHandleBackingDeviceName(first_components[0].get(), in TEST()
301 ASSERT_EQ(first_underlying_devices[0], first_device); in TEST()
571 std::string first_device = TFE_TensorHandleBackingDeviceName( in TEST() local
573 ASSERT_EQ(underlying_devices[0], first_device); in TEST()
|
| D | parallel_device_remote_test.cc | 115 const char* first_device = "/job:worker/replica:0/task:1/device:CPU:0"; in TEST() local
118 std::array<const char*, 2> underlying_devices{first_device, second_device}; in TEST()
|
| D | parallel_device_testlib.h | 111 void BasicTestsForTwoDevices(TFE_Context* context, const char* first_device,
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 330 first_device, _ = list(self._single_device_savers.items())[0]
337 with ops.device(saveable_object_util.set_cpu0(first_device)):
|