Searched refs:ThenMemcpyD2D (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | collective_permute_thunk.cc | 147 dest_stream->ThenMemcpyD2D(&dest, src, src.size()); in EnqueueCopy()
155 dest_stream->ThenWaitFor(src_stream).ThenMemcpyD2D(&dest, src, src.size()); in EnqueueCopy()
|
| D | gpu_executable.cc | 528 run_options->stream()->ThenMemcpyD2D(&result_buffer, aliased_buffer, in ExecuteAsyncOnStreamImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | local_device_state.cc | 95 transfer_stream->ThenMemcpyD2D(&dst_buffer, src_buffer, dst_buffer.size()); in ThenMemcpyDeviceToDevice()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_impl.h | 272 ->ThenMemcpyD2D(&dst_device_mem, src_device_mem,
292 ->ThenMemcpyD2D(&dst_device_mem, src_device_mem,
|
| D | tridiagonal_solve_op_gpu.cu.cc | 84 ->ThenMemcpyD2D(&dst_device_mem, src_device_mem, in CopyDeviceToDevice()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 238 stream->ThenMemcpyD2D(dynamic_input_base, static_input, in UpdateDynamicInputs()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 937 ->ThenMemcpyD2D(&batch_mean_ptr, estimated_mean_ptr, in operator ()()
948 ->ThenMemcpyD2D(&batch_var_ptr, estimated_variance_ptr, in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1763 Stream &ThenMemcpyD2D(DeviceMemoryBase *gpu_dst, in ThenMemcpyD2D() function
|