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.h718 DeviceMemory<float> *gpu_unquantized_dst);
725 DeviceMemory<float> *gpu_unquantized_dst) { in ThenMemcpyH2DQuantized() argument
728 Quantization<ElementType>::kModeId, gpu_unquantized_dst); in ThenMemcpyH2DQuantized()
733 DeviceMemory<float> *gpu_unquantized_dst);
Ddnn.h2085 DeviceMemory<float>* gpu_unquantized_dst) = 0;
Dstream.cc1277 DeviceMemory<float> *gpu_unquantized_dst) { in ThenMemcpyH2DQuantized() argument
1279 PARAM(gpu_unquantized_dst)); in ThenMemcpyH2DQuantized()
1283 gpu_unquantized_dst)); in ThenMemcpyH2DQuantized()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4617 DeviceMemory<float>* gpu_unquantized_dst) { in DoMemcpyH2DQuantized() argument
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc4578 DeviceMemory<float>* gpu_unquantized_dst) { in DoMemcpyH2DQuantized() argument