Searched refs:DeviceToDeviceCopyThunk (Results 1 – 3 of 3) sorted by relevance
54 class DeviceToDeviceCopyThunk : public Thunk {59 DeviceToDeviceCopyThunk(const BufferAllocation::Slice& source_buffer,64 DeviceToDeviceCopyThunk(const DeviceToDeviceCopyThunk&) = delete;65 DeviceToDeviceCopyThunk& operator=(const DeviceToDeviceCopyThunk&) = delete;
43 DeviceToDeviceCopyThunk::DeviceToDeviceCopyThunk( in DeviceToDeviceCopyThunk() function in xla::gpu::DeviceToDeviceCopyThunk52 Status DeviceToDeviceCopyThunk::ExecuteOnStream( in ExecuteOnStream()
507 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in HandleCustomCall()557 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in HandleTriangularSolve()1040 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in HandleScatter()1230 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in HandleSort()1440 AddThunkToThunkSequence(absl::make_unique<DeviceToDeviceCopyThunk>( in HandleAllReduce()1455 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in HandleAllReduce()1719 return absl::make_unique<DeviceToDeviceCopyThunk>( in BuildDeviceToDeviceCopyThunk()1842 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in BuildGemmThunk()