Searched defs:source_target_pairs (Results 1 – 6 of 6) sorted by relevance
111 def collective_permute(x, source_target_pairs, name=None): argument
191 repeated SourceTarget source_target_pairs = 52; field
376 const std::vector<std::pair<int64, int64>>& source_target_pairs() const { in source_target_pairs() function
401 std::vector<std::pair<int64, int64>> source_target_pairs( in CreateFromProto() local852 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CreateCollectivePermute() argument3417 HloInstruction::source_target_pairs() const { in source_target_pairs() function in xla::HloInstruction
603 const std::vector<std::pair<int64, int64>>& source_target_pairs) in HloCollectivePermuteInstruction() argument
2216 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument3253 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument