Searched defs:source_target_pairs (Results 1 – 8 of 8) sorted by relevance
| /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/mlir/xla/ |
| D | hlo_function_importer.cc | 809 source_target_pairs) { in ConvertSourceTargetPairs()
|
| D | mlir_hlo_to_hlo.cc | 156 llvm::Optional<mlir::DenseIntElementsAttr> source_target_pairs) { in Convert_source_target_pairs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 196 repeated SourceTarget source_target_pairs = 52; field
|
| D | hlo_instructions.h | 423 const std::vector<std::pair<int64, int64>>& source_target_pairs() const { in source_target_pairs() function
|
| D | hlo_instruction.cc | 430 std::vector<std::pair<int64, int64>> source_target_pairs( in CreateFromProto() local
932 const std::vector<std::pair<int64, int64>>& source_target_pairs, in CreateCollectivePermute() argument
3762 HloInstruction::source_target_pairs() const { in source_target_pairs() function in xla::HloInstruction
|
| D | hlo_instructions.cc | 654 const std::vector<std::pair<int64, int64>>& source_target_pairs, in HloCollectivePermuteInstruction() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2246 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
3323 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
|