Searched defs:source_target_pairs (Results 1 – 13 of 13) sorted by relevance
| /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 | 113 def collective_permute(x, source_target_pairs, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | collective_permute_thunk.cc | 223 std::vector<std::pair<int64, int64>> source_target_pairs, in CollectivePermuteThunk() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 200 repeated SourceTarget source_target_pairs = 52; field
|
| D | hlo_instruction.cc | 478 std::vector<std::pair<int64, int64>> source_target_pairs( in CreateFromProto() local
1075 const std::vector<std::pair<int64, int64>>& source_target_pairs, in CreateCollectivePermute() argument
1085 const std::vector<std::pair<int64, int64>>& source_target_pairs, in CreateCollectivePermuteStart() argument
4132 HloInstruction::source_target_pairs() const { in source_target_pairs() function in xla::HloInstruction
|
| D | hlo_instructions.h | 507 const std::vector<std::pair<int64, int64>>& source_target_pairs() const { in source_target_pairs() function
|
| D | hlo_instructions.cc | 770 const std::vector<std::pair<int64, int64>>& source_target_pairs, in HloCollectivePermuteInstruction() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 939 source_target_pairs, in ConvertSourceTargetPairs()
|
| D | mlir_hlo_to_hlo.cc | 172 llvm::Optional<mlir::DenseIntElementsAttr> source_target_pairs) { in Convert_source_target_pairs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 722 void* output_buffer, const void* source_target_pairs, in __xla_cpu_runtime_CollectivePermute()
|
| D | ir_emitter.cc | 1266 std::string source_target_pairs = absl::StrJoin( in HandleCollectivePermute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 821 std::vector<std::pair<int64, int64>> source_target_pairs; in ExchangeHalo() local
855 std::vector<std::pair<int64, int64>> source_target_pairs; in ExchangeHalo() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2998 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
4696 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
|