Searched defs:underlying_devices (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | composite_device.h | 41 const std::vector<string>* underlying_devices() const { in underlying_devices() function
60 const std::vector<string>& underlying_devices) in CompositeDevice()
|
| D | composite_device.cc | 26 const std::vector<string>& underlying_devices, const int unique_device_id, in MakeDevice()
36 const std::vector<string>& underlying_devices, const string& device_name, in MakeDevice()
|
| D | composite_device_test.cc | 26 std::vector<string> underlying_devices; in TEST() local
71 std::vector<string> underlying_devices; in TEST() local
|
| D | replicate_per_replica_nodes_test.cc | 89 const std::vector<string> underlying_devices = {"TPU:0", "TPU:1"}; in TEST() local
134 const std::vector<string> underlying_devices = {"TPU:0"}; in TEST() local
225 const std::vector<string> underlying_devices = {"TPU:0", "TPU:1"}; in TEST() local
303 const std::vector<string> underlying_devices = {"TPU:0", "TPU:1"}; in TEST() local
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 122 std::array<const char*, 2> underlying_devices{first_device_name, in TEST() local
184 std::array<const char*, 2> underlying_devices{ in TEST() local
315 std::array<const char*, 1> underlying_devices{ in TEST() local
434 std::array<const char*, 2> underlying_devices{ in TestCollective() local
531 std::array<const char*, 2> underlying_devices{ in TEST() local
|
| D | parallel_device_testlib.h | 155 const std::array<const char*, num_devices>& underlying_devices, in RegisterParallelDevice()
|
| D | parallel_device_remote_test.cc | 118 std::array<const char*, 2> underlying_devices{first_device, second_device}; in TEST() local
|
| D | parallel_device_testlib.cc | 210 std::array<const char*, 2> underlying_devices{first_device, second_device}; in BasicTestsForTwoDevices() local
|
| D | parallel_device.cc | 365 const char* const* underlying_devices, in AllocateParallelDevice()
|
| /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()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_test.cc | 87 std::vector<string> underlying_devices = { in TEST_F() local
|
| D | context.cc | 919 const std::vector<string>& underlying_devices, const string& device_name, in FindOrCreateCompositeDevice()
|