| /external/tensorflow/tensorflow/dtensor/proto/ |
| D | layout.proto | 56 repeated string local_devices = 5; field
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_env.h | 48 std::vector<Device*> local_devices; member
|
| D | master_env.h | 78 std::vector<Device*> local_devices; member
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 137 const std::vector<std::shared_ptr<PjRtDevice>>& local_devices() { in local_devices() function
379 const std::vector<std::shared_ptr<PjRtDevice>>& local_devices() const { in local_devices() function
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | api.py | 443 def local_devices( function
|
| D | layout.py | 388 def local_devices(self) -> List[str]: member in Mesh
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | tensor_layout.h | 138 absl::Span<const std::string> local_devices() const { return local_devices_; } in local_devices() function
|
| D | tensor_layout.cc | 255 const std::vector<std::string>& local_devices, in GetMesh()
537 std::vector<std::string> local_devices; in GenerateMeshDevicesForTests() local
603 std::vector<std::string> local_devices; in FromString() local
|
| D | dtensor_device.cc | 848 auto local_devices = target_parallel_device->mesh_config().local_devices(); in Pack() local
998 auto local_devices = target_parallel_device->mesh_config().local_devices(); in SparsePack() local
|
| D | dtensor_device_util.cc | 63 absl::Span<const std::string> local_devices = target_mesh.local_devices(); in BroadcastTensorHandleToParallelTensor() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.h | 270 std::vector<std::pair<GlobalDeviceId, int64_t>> local_devices; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_utils.cc | 203 const std::vector<GlobalDeviceId>* local_devices) { in GetNumLocalParticipants()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | move_compilation_to_host.cc | 94 absl::Span<const std::string> local_devices = mesh.local_devices(); in CreateSendRecvOpsToTransferProgramKey() local
|
| /external/intel-media-driver/cmrtlib/linux/hardware/ |
| D | drm_device.h | 1107 static void drmFoldDuplicatedDevices(drmDevicePtr local_devices[], int count) in drmFoldDuplicatedDevices()
1143 drmDevicePtr local_devices[MAX_DRM_NODES]; in drmGetDevices2() local
1223 drmDevicePtr local_devices[MAX_DRM_NODES]; in drmGetDevice2() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.cc | 731 std::vector<DeviceAttributes> local_devices; in EnableCollectiveOps() local
|
| D | context.cc | 786 std::vector<std::unique_ptr<Device>> local_devices, remote_devices; in AddDevices() local
1206 auto local_devices = local_device_mgr()->ListDevices(); in ClearResourceContainer() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/coordination/ |
| D | coordination_service_agent.cc | 322 const CoordinationServiceDeviceInfo& local_devices) { in WaitForAllTasks()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 114 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
|
| /external/libdrm/ |
| D | xf86drm.c | 4517 static void drmFoldDuplicatedDevices(drmDevicePtr local_devices[], int count) in drmFoldDuplicatedDevices()
4774 drmDevicePtr local_devices[MAX_DRM_NODES]; in drmGetDevices2() local
|