/external/tensorflow/tensorflow/core/common_runtime/ |
D | composite_device_test.cc | 26 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 …]
|
D | composite_device.cc | 26 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 …]
|
D | composite_device.h | 41 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()
|
D | replicate_per_replica_nodes_test.cc | 89 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()
|
D | process_function_library_runtime_test.cc | 871 composite_device->underlying_devices(); in TEST_F()
|
/external/tensorflow/tensorflow/python/distribute/parallel_device/ |
D | pywrap_parallel_device.cc | 44 [](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/ |
D | context_test.cc | 87 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()
|
D | tensor_handle_test.cc | 195 EXPECT_EQ(device->underlying_devices()->size(), 4); in TEST_F() 242 EXPECT_EQ(device->underlying_devices()->size(), 1); in TEST_F()
|
D | context.cc | 919 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()
|
D | context.h | 489 const std::vector<string>& underlying_devices, const string& device_name,
|
D | execute.cc | 435 composite_device->underlying_devices(); in GetOrCreateKernelAndDevice()
|
/external/tensorflow/tensorflow/c/eager/parallel_device/ |
D | parallel_device_testlib.h | 106 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()
|
D | parallel_device_test.cc | 122 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 …]
|
D | parallel_device_testlib.cc | 210 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()
|
D | parallel_device.h | 58 const char* const* underlying_devices,
|
D | parallel_device_remote_test.cc | 118 std::array<const char*, 2> underlying_devices{first_device, second_device}; in TEST() local 119 RegisterParallelDevice(context.get(), device_name, underlying_devices, in TEST()
|
D | parallel_device.cc | 365 const char* const* underlying_devices, in AllocateParallelDevice() argument 376 underlying_devices_vector.push_back(underlying_devices[device_index]); in AllocateParallelDevice()
|