Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream.h695 DeviceMemory<float> *gpu_unquantized_dst);
702 DeviceMemory<float> *gpu_unquantized_dst) { in ThenMemcpyH2DQuantized() argument
705 Quantization<ElementType>::kModeId, gpu_unquantized_dst); in ThenMemcpyH2DQuantized()
710 DeviceMemory<float> *gpu_unquantized_dst);
Ddnn.h2019 DeviceMemory<float>* gpu_unquantized_dst) = 0;
Dstream.cc1723 DeviceMemory<float> *gpu_unquantized_dst) { in ThenMemcpyH2DQuantized() argument
1725 PARAM(gpu_unquantized_dst)); in ThenMemcpyH2DQuantized()
1730 gpu_unquantized_dst)); in ThenMemcpyH2DQuantized()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4084 DeviceMemory<float>* gpu_unquantized_dst) { in DoMemcpyH2DQuantized() argument
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3981 DeviceMemory<float>* gpu_unquantized_dst) { in DoMemcpyH2DQuantized() argument