Home
last modified time | relevance | path

Searched defs:conv_input_data (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1145 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
1163 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
1182 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
1203 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
2489 const DeviceMemory<float>& conv_input_data, in DoFusedConvolutionBiasActivation()
Dstream.cc448 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in FusedConvolveWithAlgorithm()
479 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm()
510 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm()
542 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm()
573 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3179 const DeviceMemory<ElementType>& conv_input_data, in DoFusedConvolveImpl()
3758 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3780 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3802 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3825 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3856 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3677 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3693 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3709 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3726 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
4620 const DeviceMemory<T>& conv_input_data, in DoFusedConvolutionBiasActivationImpl()
4708 const DeviceMemory<float>& conv_input_data, in DoFusedConvolutionBiasActivation()