Searched refs:coefficient_scales (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 357 const DeviceMemory<float>& coefficient_scales, in DoConvolveQuantized() argument
370 const DeviceMemory<float>& coefficient_scales, in DoConvolveQuantized() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 381 const DeviceMemory<float>& coefficient_scales, in DoConvolveQuantized() argument
394 const DeviceMemory<float>& coefficient_scales, in DoConvolveQuantized() argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1354 const DeviceMemory<float>& coefficient_scales,
1365 const DeviceMemory<float>& coefficient_scales,
|
| D | stream.cc | 626 const DeviceMemory<float> &coefficient_scales, in ThenConvolveQuantized() argument
632 PARAM(coefficient_scales), PARAM(convolution_descriptor), in ThenConvolveQuantized()
638 filter_coefficients, coefficient_scales, convolution_descriptor, in ThenConvolveQuantized()
653 const DeviceMemory<float> &coefficient_scales, in ThenConvolveQuantized() argument
659 PARAM(coefficient_scales), PARAM(convolution_descriptor), in ThenConvolveQuantized()
665 filter_coefficients, coefficient_scales, convolution_descriptor, in ThenConvolveQuantized()
|
| D | stream.h | 310 const DeviceMemory<float> &coefficient_scales,
320 const DeviceMemory<float> &coefficient_scales,
|