| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_collective_permute_thunk.cc | 57 const std::vector<std::pair<int64, int64>> source_target_pairs = in GetNcclCollectivePermuteConfig() local
78 const std::vector<std::pair<int64, int64>> source_target_pairs = in IsDegenerate() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | cross_replica_ops.cc | 114 std::vector<std::pair<int64, int64>> source_target_pairs(num_pairs); in Compile() local
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 112 def collective_permute(x, source_target_pairs, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 505 std::vector<std::pair<int64_t, int64_t>> source_target_pairs( in CreateFromProto() local
1227 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs, in CreateCollectivePermute() argument
1238 absl::Span<const std::pair<int64_t, int64_t>> source_target_pairs, in CreateCollectivePermute() argument
1249 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs, in CreateCollectivePermuteStart() argument
1260 absl::Span<const std::pair<int64_t, int64_t>> source_target_pairs, in CreateCollectivePermuteStart() argument
4430 HloInstruction::source_target_pairs() const { in source_target_pairs() function in xla::HloInstruction
|
| D | hlo.proto | 235 repeated SourceTarget source_target_pairs = 52; field
|
| D | hlo_instructions.h | 557 const std::vector<std::pair<int64_t, int64_t>>& source_target_pairs() const { in source_target_pairs() function
|
| D | hlo_instructions.cc | 826 const std::vector<std::pair<int64, int64>>& source_target_pairs, in HloCollectivePermuteInstruction() argument
837 absl::Span<const std::pair<int64_t, int64_t>> source_target_pairs, in HloCollectivePermuteInstruction() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 984 source_target_pairs, in ConvertSourceTargetPairs()
|
| D | mlir_hlo_to_hlo.cc | 175 llvm::Optional<mlir::DenseIntElementsAttr> source_target_pairs) { in Convert_source_target_pairs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 744 void* output_buffer, const void* source_target_pairs, in __xla_cpu_runtime_CollectivePermute()
|
| D | ir_emitter.cc | 1268 std::string source_target_pairs = absl::StrJoin( in HandleCollectivePermute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 832 std::vector<std::pair<int64, int64>> source_target_pairs; in ExchangeHalo() local
894 std::vector<std::pair<int64, int64>> source_target_pairs; in ExchangeHalo() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 3087 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
4558 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
|