Searched refs:CollectivePermuteThunk (Results 1 – 3 of 3) sorted by relevance
28 class CollectivePermuteThunk : public Thunk {30 CollectivePermuteThunk(
221 CollectivePermuteThunk::CollectivePermuteThunk( in CollectivePermuteThunk() function in xla::gpu::CollectivePermuteThunk230 Status CollectivePermuteThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
3095 AddThunkToThunkSequence(absl::make_unique<CollectivePermuteThunk>( in EmitCollectivePermuteFromMlir()