Home
last modified time | relevance | path

Searched defs:num_global_devices (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dcollective_communicator.cc84 const int num_global_devices = col_params->group.group_size; in Enqueue() local
H A Dnccl_manager.h151 int num_global_devices; member
H A Dnccl_manager.cc193 const int num_global_devices; // devices across all nodes member
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/
H A Dapi.py483 def num_global_devices(device_type: str) -> int: function