| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/ |
| H A D | parallel_device_test.cc | 126 std::array<const char*, 2> underlying_devices{first_device_name, in TEST() local
167 std::array<const char*, 2> underlying_devices{ in TEST() local
308 std::array<const char*, 1> underlying_devices{ in TEST() local
426 std::array<const char*, 2> underlying_devices{ in TestCollective() local
522 std::array<const char*, 2> underlying_devices{ in TEST() local
587 std::array<const char*, 2> underlying_devices{ in TEST() local
|
| H A D | parallel_device_testlib.h | 155 const std::array<const char*, num_devices>& underlying_devices, in RegisterParallelDevice()
|
| H A D | parallel_device_lib.h | 86 const std::vector<std::string>& underlying_devices() const { in underlying_devices() function
|
| H A D | parallel_device_remote_test.cc | 118 std::array<const char*, 2> underlying_devices{first_device, second_device}; in TEST() local
|
| H A D | parallel_device_testlib.cc | 210 std::array<const char*, 2> underlying_devices{first_device, second_device}; in BasicTestsForTwoDevices() local
|
| H A D | parallel_device.cc | 408 const char* const* underlying_devices, in AllocateParallelDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A 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
221 const std::vector<string> underlying_devices = {"TPU:0", "TPU:1"}; in TEST() local
299 const std::vector<string> underlying_devices = {"TPU:0", "TPU:1"}; in TEST() local
|
| H A 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()
|
| H A 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()
|
| H A D | composite_device_test.cc | 26 std::vector<string> underlying_devices; in TEST() local
71 std::vector<string> underlying_devices; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| H A D | pywrap_parallel_device.cc | 44 [](const char* name, std::vector<std::string> underlying_devices) { in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | context_test.cc | 82 std::vector<string> underlying_devices = { in TEST_F() local
|
| H A D | context.cc | 1152 const std::vector<string>& underlying_devices, const string& device_name, in FindOrCreateCompositeDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device.cc | 2033 std::vector<std::string> underlying_devices; in AddMesh() local
|