Home
last modified time | relevance | path

Searched refs:is_per_channel (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc225 const bool is_per_channel = affine_quantization->scale->size > 1; in PopulateConvolutionQuantizationParams() local
226 if (is_per_channel) { in PopulateConvolutionQuantizationParams()
244 const float scale = is_per_channel ? filter_scales[i] : filter_scales[0]; in PopulateConvolutionQuantizationParams()
Dconv.cc219 bool is_per_channel, in AllocateTemporaryTensorsIfRequired() argument
289 if (is_per_channel) { in AllocateTemporaryTensorsIfRequired()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc49 bool is_per_channel; member
444 if (tensor_pair.second.is_per_channel) { in QuantizeWeightsInt8()
Dquantize_model.cc143 bool is_per_channel, int channel_dim_index, in QuantizeBias() argument
155 if (is_per_channel) { in QuantizeBias()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_test.cc1063 bool is_per_channel = false, in SetupMatrixVectorData() argument
1117 if (is_per_channel) { in SetupMatrixVectorData()
1162 bool is_per_channel = true) { in TestPerChannelDotprodMatrixBatchVectorMultiply() argument
1164 SetupMatrixVectorData(rows, cols, batch, negative, is_per_channel); in TestPerChannelDotprodMatrixBatchVectorMultiply()