Home
last modified time | relevance | path

Searched refs:HostToDeviceCopyThunk (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcopy_thunk.h31 class HostToDeviceCopyThunk : public Thunk {
36 HostToDeviceCopyThunk(ThunkInfo thunk_info, const void* source_address,
40 HostToDeviceCopyThunk(const HostToDeviceCopyThunk&) = delete;
41 HostToDeviceCopyThunk& operator=(const HostToDeviceCopyThunk&) = delete;
Dcopy_thunk.cc24 HostToDeviceCopyThunk::HostToDeviceCopyThunk( in HostToDeviceCopyThunk() function in xla::gpu::HostToDeviceCopyThunk
32 Status HostToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()