Home
last modified time | relevance | path

Searched refs:destination_buffer_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcopy_thunk.cc30 destination_buffer_(destination_buffer), in HostToDeviceCopyThunk()
37 buffer_allocations.GetDeviceAddress(destination_buffer_); in ExecuteOnStream()
49 destination_buffer_(destination_buffer), in DeviceToDeviceCopyThunk()
56 buffer_allocations.GetDeviceAddress(destination_buffer_); in ExecuteOnStream()
Dcopy_thunk.h49 const BufferAllocation::Slice destination_buffer_; variable
73 const BufferAllocation::Slice destination_buffer_; variable
Dnccl_all_reduce_thunk.h56 const BufferAllocation::Slice destination_buffer_; variable
Dnccl_all_reduce_thunk.cc326 buffer_allocations.GetDeviceAddress(destination_buffer_); in ExecuteOnStream()
353 destination_buffer_(destination_buffer) {} in NcclAllReduceThunk()