Searched defs:conv_input_data (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2504 const DeviceMemory<Type>& conv_input_data, ScaleType conv_input_scale, in DoFusedConvolveImpl()
2970 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
2992 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3014 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3037 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 417 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithScratch()
450 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithScratch()
484 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithScratch()
571 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
609 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
648 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
748 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolve()
|