Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcollective_ops_utils_test.cc66 std::optional<bool> use_global_device_ids; member
123 std::optional<bool> use_global_device_ids; member
H A Dcollective_ops_utils.cc96 bool has_channel_id, std::optional<bool> use_global_device_ids) { in GetCollectiveOpGroupMode()
H A Dhlo_instruction.cc1261 bool use_global_device_ids) { in CreateAllGather()
1272 bool use_global_device_ids) { in CreateAllGatherStart()
1282 const std::optional<int64_t>& channel_id, bool use_global_device_ids) { in CreateAllReduce()
1293 const std::optional<int64_t>& channel_id, bool use_global_device_ids, in CreateReduceScatter()
1305 const std::optional<int64_t>& channel_id, bool use_global_device_ids) { in CreateAllReduceStart()
H A Dhlo_instructions.h583 bool use_global_device_ids() const { return use_global_device_ids_; } in use_global_device_ids() function
638 bool use_global_device_ids() const { return use_global_device_ids_; } in use_global_device_ids() function
H A Dhlo.proto239 bool use_global_device_ids = 71; field
H A Dhlo_instructions.cc789 const std::optional<int64_t>& channel_id, bool use_global_device_ids) in HloAllGatherInstruction()
838 const std::optional<int64_t>& channel_id, bool use_global_device_ids) in HloAllReduceInstructionBase()
900 const std::optional<int64_t>& channel_id, bool use_global_device_ids, in HloReduceScatterInstruction()
H A Dhlo_parser.cc1433 optional<bool> use_global_device_ids; in CreateInstruction() local
1470 optional<bool> use_global_device_ids; in CreateInstruction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnccl_collective_thunk.h79 OpT op, std::optional<bool> use_global_device_ids) { in GetNcclCollectiveConfigForMlir()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_runtime.cc696 std::optional<bool> use_global_device_ids, int64_t op_id) { in GetRendezvousKey()
763 int32_t channel_id_present, int32_t use_global_device_ids, int64_t op_id, in __xla_cpu_runtime_AllReduce()
H A Dir_emitter.cc1250 bool use_global_device_ids = in HandleAllReduceMultipleReplica() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc2873 const std::optional<bool> use_global_device_ids) { in AllGather()
2940 const std::optional<bool> use_global_device_ids) { in AllReduce()
3022 const std::optional<bool> use_global_device_ids) { in ReduceScatter()
4658 const std::optional<bool> use_global_device_ids) { in AllGather()
4673 const std::optional<bool> use_global_device_ids) { in AllReduce()
4684 const std::optional<bool> use_global_device_ids) { in ReduceScatter()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dmlir_hlo_to_hlo.cc199 llvm::Optional<bool> use_global_device_ids) { in Convert_use_global_device_ids()