| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | nccl_collective_permute_thunk.cc | 64 const std::vector<std::pair<int64_t, int64_t>> source_target_pairs = in GetNcclCollectivePermuteConfig() local
85 const std::vector<std::pair<int64_t, int64_t>> source_target_pairs = in IsDegenerate() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | cross_replica_ops.cc | 114 std::vector<std::pair<int64_t, int64_t>> source_target_pairs(num_pairs); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instruction.cc | 561 std::vector<std::pair<int64_t, int64_t>> source_target_pairs( in CreateFromProto() local
1324 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs, in CreateCollectivePermute() argument
1335 absl::Span<const std::pair<int64_t, int64_t>> source_target_pairs, in CreateCollectivePermute() argument
1346 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs, in CreateCollectivePermuteStart() argument
1357 absl::Span<const std::pair<int64_t, int64_t>> source_target_pairs, in CreateCollectivePermuteStart() argument
4705 HloInstruction::source_target_pairs() const { in source_target_pairs() function in xla::HloInstruction
|
| H A D | hlo.proto | 255 repeated SourceTarget source_target_pairs = 52; field
|
| H A D | hlo_instructions.h | 771 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs() const { in source_target_pairs() function
|
| H A D | hlo_instructions.cc | 990 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs, in HloCollectivePermuteInstruction() argument
1001 absl::Span<const std::pair<int64_t, int64_t>> source_target_pairs, in HloCollectivePermuteInstruction() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ops/ |
| H A D | tpu_ops.py | 108 def collective_permute(x, source_target_pairs, name=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_util.cc | 748 std::vector<std::pair<int64_t, int64_t>> source_target_pairs; in ExchangeHalo() local
810 std::vector<std::pair<int64_t, int64_t>> source_target_pairs; in ExchangeHalo() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_runtime.cc | 833 const void* source_target_pairs, int32_t source_target_pairs_size) { in __xla_cpu_runtime_CollectivePermute()
|
| H A D | ir_emitter.cc | 1345 std::string source_target_pairs = absl::StrJoin( in HandleCollectivePermute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | hlo_function_importer.cc | 1665 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs, in ConvertSourceTargetPairs() argument
|
| H A D | mlir_hlo_to_hlo.cc | 205 llvm::Optional<mlir::DenseIntElementsAttr> source_target_pairs) { in Convert_source_target_pairs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | lmhlo_to_jitrt.cc | 1249 std::vector<std::pair<int64_t, int64_t>> source_target_pairs = in SetSpecificAttrs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 3228 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs) { in CollectivePermute() argument
4716 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs) { in CollectivePermute() argument
|