Searched refs:weight_scales (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | subgraph_quantizer.cc | 155 const std::vector<float>& weight_scales = weight_tensor->quantization->scale; in SymmetricPerChannelBiasQuantize() local
157 if (weight_scales.size() != channel_dim_size) { in SymmetricPerChannelBiasQuantize()
159 weight_scales.size()); in SymmetricPerChannelBiasQuantize()
166 scales[i] = input_tensor->quantization->scale[0] * weight_scales[i]; in SymmetricPerChannelBiasQuantize()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 398 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized() argument
408 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 389 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized() argument
399 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized() argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1511 const DeviceMemory<float>& weight_scales,
1523 const DeviceMemory<float>& weight_scales,
|
| D | stream.cc | 1160 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized() argument
1164 VLOG_CALL(PARAM(input_data), PARAM(weights), PARAM(weight_scales), in ThenMatMulQuantized()
1171 weight_scales, input_dimensions, in ThenMatMulQuantized()
1182 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized() argument
1186 VLOG_CALL(PARAM(input_data), PARAM(weights), PARAM(weight_scales), in ThenMatMulQuantized()
1193 weight_scales, input_dimensions, in ThenMatMulQuantized()
|
| D | stream.h | 500 const DeviceMemory<float> &weight_scales,
507 const DeviceMemory<float> &weight_scales,
|