Home
last modified time | relevance | path

Searched refs:source_target_pairs_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc226 source_target_pairs_(std::move(source_target_pairs)), in CollectivePermuteThunk()
250 for (const auto& src_dest : source_target_pairs_) { in ExecuteOnStream()
265 if (absl::c_none_of(source_target_pairs_, in ExecuteOnStream()
Dcollective_permute_thunk.h38 const std::vector<std::pair<int64, int64>> source_target_pairs_;
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h508 return source_target_pairs_; in source_target_pairs()
527 const std::vector<std::pair<int64, int64>> source_target_pairs_; variable
Dhlo_instructions.cc773 source_target_pairs_(source_target_pairs) { in HloCollectivePermuteInstruction()