Searched refs:coefficient_scales (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 338 const DeviceMemory<float>& coefficient_scales, in DoConvolveQuantized() argument
351 const DeviceMemory<float>& coefficient_scales, in DoConvolveQuantized() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 329 const DeviceMemory<float>& coefficient_scales, in DoConvolveQuantized() argument
342 const DeviceMemory<float>& coefficient_scales, in DoConvolveQuantized() argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1288 const DeviceMemory<float>& coefficient_scales,
1299 const DeviceMemory<float>& coefficient_scales,
|
| D | stream.cc | 739 const DeviceMemory<float> &coefficient_scales, in ThenConvolveQuantized() argument
745 PARAM(coefficient_scales), PARAM(convolution_descriptor), in ThenConvolveQuantized()
752 filter_coefficients, coefficient_scales, convolution_descriptor, in ThenConvolveQuantized()
769 const DeviceMemory<float> &coefficient_scales, in ThenConvolveQuantized() argument
775 PARAM(coefficient_scales), PARAM(convolution_descriptor), in ThenConvolveQuantized()
782 filter_coefficients, coefficient_scales, convolution_descriptor, in ThenConvolveQuantized()
|
| D | stream.h | 289 const DeviceMemory<float> &coefficient_scales,
299 const DeviceMemory<float> &coefficient_scales,
|