Searched defs:destination_buffer (Results 1 – 5 of 5) sorted by relevance
26 const BufferAllocation::Slice& destination_buffer, uint64 mem_size, in HostToDeviceCopyThunk()45 const BufferAllocation::Slice& destination_buffer, uint64 mem_size, in DeviceToDeviceCopyThunk()
347 const BufferAllocation::Slice& destination_buffer, in NcclAllReduceThunk()
555 auto destination_buffer = GetAllocationSlice(*hlo); in HandleTriangularSolve() local1038 auto destination_buffer = GetAllocationSlice(*scatter); in HandleScatter() local1225 auto destination_buffer = GetAllocationSlice(*sort, shape_index); in HandleSort() local
3234 Buffer destination_buffer(gl); in iterate() local3505 Buffer destination_buffer(gl); in iterate() local
543 auto destination_buffer = GetAllocationSlice(*sort, shape_index); in HandleSort() local