Searched defs:use_global_device_ids (Results 1 – 8 of 8) sorted by relevance
155 bool use_global_device_ids; member
395 bool use_global_device_ids() const { return use_global_device_ids_; } in use_global_device_ids() function442 bool use_global_device_ids() const { return use_global_device_ids_; } in use_global_device_ids() function
184 bool use_global_device_ids = 71; field
196 bool use_global_device_ids) { in CheckReplicaGroups()
1202 optional<bool> use_global_device_ids; in ParseInstructionRhs() local1231 optional<bool> use_global_device_ids; in ParseInstructionRhs() local
1046 const absl::optional<int64>& channel_id, bool use_global_device_ids) { in CreateAllGather()1056 const absl::optional<int64>& channel_id, bool use_global_device_ids) { in CreateAllReduce()
621 const absl::optional<int64>& channel_id, bool use_global_device_ids) in HloAllGatherInstruction()669 const absl::optional<int64>& channel_id, bool use_global_device_ids) in HloAllReduceInstruction()
2719 const absl::optional<bool> use_global_device_ids) { in AllGather()4658 const absl::optional<bool> use_global_device_ids) { in AllGather()