Searched defs:conv_input_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3147 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3164 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3181 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3199 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
4024 const DeviceMemory<T>& conv_input_data, in DoFusedConvolutionBiasActivationImpl()
4112 const DeviceMemory<float>& conv_input_data, in DoFusedConvolutionBiasActivation()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2924 const DeviceMemory<ElementType>& conv_input_data, in DoFusedConvolveImpl()
3345 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3369 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3393 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3418 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 442 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in ThenFusedConvolveWithAlgorithm()
480 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
518 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
557 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
|