Home
last modified time | relevance | path

Searched defs:local_devices (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_env.h48 std::vector<Device*> local_devices; member
Dmaster_env.h75 std::vector<Device*> local_devices; member
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h112 const std::vector<std::shared_ptr<PjRtDevice>>& local_devices() { in local_devices() function
330 const std::vector<std::shared_ptr<PjRtDevice>>& local_devices() const { in local_devices() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h180 std::vector<std::pair<GlobalDeviceId, int64>> local_devices; member
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_utils.cc260 const std::vector<GlobalDeviceId>* local_devices) { in GetLocalParticipants()
/external/tensorflow/tensorflow/core/common_runtime/
Dpermuter_test.cc137 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
Dring_gatherer_test.cc145 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
Dring_reducer_test.cc168 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
Dhierarchical_tree_broadcaster_test.cc230 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc113 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc973 auto local_devices = local_device_mgr()->ListDevices(); in ClearResourceContainer() local
/external/libdrm/
Dxf86drm.c4017 static void drmFoldDuplicatedDevices(drmDevicePtr local_devices[], int count) in drmFoldDuplicatedDevices()
4234 drmDevicePtr local_devices[MAX_DRM_NODES]; in drmGetDevices2() local