Searched refs:InfeedThunk (Results 1 – 3 of 3) sorted by relevance
33 class InfeedThunk : public Thunk {37 InfeedThunk(const ShapeTree<BufferAllocation::Slice>& infeed_slices,40 InfeedThunk(const InfeedThunk&) = delete;41 InfeedThunk& operator=(const InfeedThunk&) = delete;
25 InfeedThunk::InfeedThunk( in InfeedThunk() function in xla::gpu::InfeedThunk30 Status InfeedThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
1739 return absl::make_unique<InfeedThunk>(slices, inst); in BuildInfeedThunk()