Searched refs:DeviceToDeviceCopyThunk (Results 1 – 4 of 4) sorted by relevance
52 class DeviceToDeviceCopyThunk : public Thunk {57 DeviceToDeviceCopyThunk(ThunkInfo thunk_info,62 DeviceToDeviceCopyThunk(const DeviceToDeviceCopyThunk&) = delete;63 DeviceToDeviceCopyThunk& operator=(const DeviceToDeviceCopyThunk&) = delete;
41 DeviceToDeviceCopyThunk::DeviceToDeviceCopyThunk( in DeviceToDeviceCopyThunk() function in xla::gpu::DeviceToDeviceCopyThunk49 Status DeviceToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
43 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in BuildGemmThunk()
1279 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in EmitGemmThunkFromMlir()1509 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in EmitCholeskyThunkFromMlir()1629 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in EmitTriangularSolveFromMlir()2100 AddThunkToThunkSequence(absl::make_unique<DeviceToDeviceCopyThunk>( in EmitCopyForMlir()2511 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in EmitScatterFromMlir()2882 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in EmitSortFromMlir()3157 thunks.push_back(absl::make_unique<DeviceToDeviceCopyThunk>( in EmitNcclThunkFromMlir()