Searched defs:num_local_devices (Results 1 – 10 of 10) sorted by relevance
85 const int num_local_devices = col_params->group.num_devices_per_task.at( in Enqueue() local
148 int num_local_devices; member
192 const int num_local_devices; // devices local to this node member
477 def num_local_devices(device_type: str) -> int: function
401 def num_local_devices(self) -> int: member in Mesh
74 size_t num_local_devices = replica_ids_by_host.begin()->second.size(); in TryDecomposeReplicaGroup() local
125 int num_local_devices = host.Cores(kTensorCore).size(); in GetDefaultDeviceAssignment() local
448 int num_local_devices = 0; in CreateSession() local
682 const int num_local_devices = local_devices_.size(); in ExecuteOnLocalDevices() local
64 const int num_local_devices = local_devices.size(); in BroadcastTensorHandleToParallelTensor() local