Home
last modified time | relevance | path

Searched refs:underlying_devices (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcomposite_device_test.cc26 std::vector<string> underlying_devices; in TEST() local
30 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/0, in TEST()
41 underlying_devices.push_back( in TEST()
43 underlying_devices.push_back( in TEST()
46 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/0, in TEST()
50 EXPECT_EQ(underlying_devices, *composite_device->underlying_devices()); in TEST()
55 underlying_devices.push_back( in TEST()
58 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/1, in TEST()
71 std::vector<string> underlying_devices; in TEST() local
72 underlying_devices.push_back("/job:worker/replica:0/task:0/device:CPU:0"); in TEST()
[all …]
Dcomposite_device.cc26 const std::vector<string>& underlying_devices, const int unique_device_id, in MakeDevice() argument
32 return CompositeDevice::MakeDevice(underlying_devices, device_name, status); in MakeDevice()
36 const std::vector<string>& underlying_devices, const string& device_name, in MakeDevice() argument
38 if (underlying_devices.empty()) { in MakeDevice()
44 if (!DeviceNameUtils::ParseFullName(underlying_devices.at(0), &parsed_name)) { in MakeDevice()
46 "Cannot parse device name ", underlying_devices.at(0), in MakeDevice()
51 for (int i = 1; i < underlying_devices.size(); ++i) { in MakeDevice()
53 if (!DeviceNameUtils::ParseFullName(underlying_devices.at(i), &name)) { in MakeDevice()
55 "Cannot parse device name ", underlying_devices.at(i), in MakeDevice()
62 " from device: ", underlying_devices.at(i), in MakeDevice()
[all …]
Dcomposite_device.h41 const std::vector<string>* underlying_devices() const { in underlying_devices() function
48 const std::vector<string>& underlying_devices, const int unique_device_id,
53 const std::vector<string>& underlying_devices, const string& device_name,
60 const std::vector<string>& underlying_devices) in CompositeDevice() argument
62 underlying_devices_(underlying_devices) {} in CompositeDevice()
Dreplicate_per_replica_nodes_test.cc89 const std::vector<string> underlying_devices = {"TPU:0", "TPU:1"}; in TEST() local
91 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST()
134 const std::vector<string> underlying_devices = {"TPU:0"}; in TEST() local
136 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST()
225 const std::vector<string> underlying_devices = {"TPU:0", "TPU:1"}; in TEST() local
227 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST()
303 const std::vector<string> underlying_devices = {"TPU:0", "TPU:1"}; in TEST() local
305 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST()
Dprocess_function_library_runtime_test.cc871 composite_device->underlying_devices(); in TEST_F()
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dpywrap_parallel_device.cc44 [](const char* name, std::vector<std::string> underlying_devices) { in PYBIND11_MODULE() argument
46 underlying_devices_c.reserve(underlying_devices.size()); in PYBIND11_MODULE()
47 for (const std::string& element : underlying_devices) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_test.cc87 std::vector<string> underlying_devices = { in TEST_F() local
91 TF_ASSERT_OK(context()->FindOrCreateCompositeDevice(underlying_devices, in TEST_F()
101 TF_ASSERT_OK(context()->FindOrCreateCompositeDevice(underlying_devices, in TEST_F()
105 underlying_devices.push_back("/job:worker/replica:0/task:0/device:CPU:2"); in TEST_F()
107 TF_ASSERT_OK(context()->FindOrCreateCompositeDevice(underlying_devices, in TEST_F()
Dtensor_handle_test.cc195 EXPECT_EQ(device->underlying_devices()->size(), 4); in TEST_F()
242 EXPECT_EQ(device->underlying_devices()->size(), 1); in TEST_F()
Dcontext.cc919 const std::vector<string>& underlying_devices, const string& device_name, in FindOrCreateCompositeDevice() argument
926 const uint64 hash_key = Fingerprint64(absl::StrJoin(underlying_devices, ",")); in FindOrCreateCompositeDevice()
940 device = CompositeDevice::MakeDevice(underlying_devices, in FindOrCreateCompositeDevice()
944 device = CompositeDevice::MakeDevice(underlying_devices, device_name, &s); in FindOrCreateCompositeDevice()
Dcontext.h489 const std::vector<string>& underlying_devices, const string& device_name,
Dexecute.cc435 composite_device->underlying_devices(); in GetOrCreateKernelAndDevice()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.h106 const std::array<const char*, num_devices>& underlying_devices,
155 const std::array<const char*, num_devices>& underlying_devices, in RegisterParallelDevice() argument
160 device_name, underlying_devices.data(), underlying_devices.size(), in RegisterParallelDevice()
Dparallel_device_test.cc122 std::array<const char*, 2> underlying_devices{first_device_name, in TEST() local
124 RegisterParallelDevice(context.get(), device_name, underlying_devices, in TEST()
154 ASSERT_EQ(underlying_devices[0], first_device); in TEST()
157 ASSERT_EQ(underlying_devices[1], second_device); in TEST()
184 std::array<const char*, 2> underlying_devices{ in TEST() local
187 RegisterParallelDevice(context.get(), device_name, underlying_devices, in TEST()
315 std::array<const char*, 1> underlying_devices{ in TEST() local
317 RegisterParallelDevice(context.get(), device_name, underlying_devices, in TEST()
434 std::array<const char*, 2> underlying_devices{ in TestCollective() local
437 RegisterParallelDevice(context.get(), device_name, underlying_devices, in TestCollective()
[all …]
Dparallel_device_testlib.cc210 std::array<const char*, 2> underlying_devices{first_device, second_device}; in BasicTestsForTwoDevices() local
211 RegisterParallelDevice(context, device_name, underlying_devices, in BasicTestsForTwoDevices()
249 ASSERT_EQ(underlying_devices[0], first_device); in BasicTestsForTwoDevices()
252 ASSERT_EQ(underlying_devices[1], second_device); in BasicTestsForTwoDevices()
279 ASSERT_EQ(underlying_devices[0], first_device); in BasicTestsForTwoDevices()
282 ASSERT_EQ(underlying_devices[1], second_device); in BasicTestsForTwoDevices()
Dparallel_device.h58 const char* const* underlying_devices,
Dparallel_device_remote_test.cc118 std::array<const char*, 2> underlying_devices{first_device, second_device}; in TEST() local
119 RegisterParallelDevice(context.get(), device_name, underlying_devices, in TEST()
Dparallel_device.cc365 const char* const* underlying_devices, in AllocateParallelDevice() argument
376 underlying_devices_vector.push_back(underlying_devices[device_index]); in AllocateParallelDevice()