Searched defs:multi_device (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | process_function_library_runtime.cc | 301 bool multi_device; in HasMultiDeviceHandle() local
1745 bool multi_device = HasMultiDeviceHandle(handle); in Run() local
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_ucc.py | 357 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
|
| H A D | test_c10d_gloo.py | 1499 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
|
| H A D | test_c10d_common.py | 753 multi_device=False, argument
|
| H A D | test_c10d_nccl.py | 862 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
|