Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc270 stream->ThenMemcpyH2D<int8>(*padded_data, &typed_new_input_memory); in UpdateDynamicInputs()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h1747 Stream &ThenMemcpyH2D(port::ArraySlice<T> host_src, in ThenMemcpyH2D() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4574 stream->ThenMemcpyH2D<float>(output_host, output_data); in DoDepthConcatenate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc4534 stream->ThenMemcpyH2D<float>(output_host, output_data); in DoDepthConcatenate()