Searched refs:OutfeedThunk (Results 1 – 3 of 3) sorted by relevance
31 class OutfeedThunk : public Thunk {35 OutfeedThunk(ThunkInfo thunk_info, std::vector<ShapedSlice> source_slices);37 OutfeedThunk(const OutfeedThunk&) = delete;38 OutfeedThunk& operator=(const OutfeedThunk&) = delete;
28 OutfeedThunk::OutfeedThunk(ThunkInfo thunk_info, in OutfeedThunk() function in xla::gpu::OutfeedThunk33 Status OutfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
3223 AddThunkToThunkSequence(absl::make_unique<OutfeedThunk>( in HandleOutfeed()