Home
last modified time | relevance | path

Searched refs:ThenMemcpyH2D (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dconvolution_thunk.cc66 stream->ThenMemcpyH2D<void*>(ptrs, &tuple_addr); in ExecuteOnStream()
Dcudnn_batchnorm_thunk.cc206 stream->ThenMemcpyH2D<void*>(ptrs, &tuple_addr); in ExecuteOnStream()
282 stream->ThenMemcpyH2D<void*>(ptrs, &tuple_addr); in ExecuteOnStream()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h1705 Stream &ThenMemcpyH2D(port::ArraySlice<T> host_src, in ThenMemcpyH2D() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4041 stream->ThenMemcpyH2D<float>(output_host, output_data); in DoDepthConcatenate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3937 stream->ThenMemcpyH2D<float>(output_host, output_data); in DoDepthConcatenate()