Searched refs:TupleThunk (Results 1 – 3 of 3) sorted by relevance
35 class TupleThunk : public Thunk {37 TupleThunk(absl::Span<const BufferAllocation::Slice> tuple_element_buffers, in TupleThunk() function45 TupleThunk(const TupleThunk&) = delete;46 TupleThunk& operator=(const TupleThunk&) = delete;
25 Status TupleThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
766 AddThunkToThunkSequence(absl::make_unique<TupleThunk>( in HandleTuple()1462 thunks.push_back(absl::make_unique<TupleThunk>( in HandleAllReduce()