Searched defs:side_input_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2928 const DeviceMemory<ElementType>& side_input_data, in DoFusedConvolveImpl()
3349 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3373 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3397 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3422 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3151 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3168 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3185 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3203 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 446 const DeviceMemory<double> &side_input_data, double side_input_scale, in ThenFusedConvolveWithAlgorithm()
484 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
522 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
561 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
|