Searched defs:side_input_data (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2508 const DeviceMemory<Type>& side_input_data, ScaleType side_input_scale, in DoFusedConvolveImpl()
2974 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
2996 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3018 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3041 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 421 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithScratch()
454 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithScratch()
488 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithScratch()
575 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
613 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
652 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
752 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolve()
|