Searched refs:channel_dim_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 136 int32_t channel_dim_index, in FillPerChannelMinMax() argument
151 if (channel_dim_index > 3) { in FillPerChannelMinMax()
155 channel_dim_index); in FillPerChannelMinMax()
158 const int32_t channel_dim_size = dimension[channel_dim_index]; in FillPerChannelMinMax()
159 quantization_params->quantized_dimension = channel_dim_index; in FillPerChannelMinMax()
172 int channel_idx = indices[channel_dim_index]; in FillPerChannelMinMax()
293 int32_t channel_dim_index, in SymmetricPerChannelQuantization() argument
301 const int32_t channel_dim_size = tensor->shape[channel_dim_index]; in SymmetricPerChannelQuantization()
308 FillPerChannelMinMax(input, tensor->shape, channel_dim_index, in SymmetricPerChannelQuantization()
329 channel_dim_index, output_value); in SymmetricPerChannelQuantization()
[all …]
|
| D | quantization_utils.h | 46 int32_t channel_dim_index,
64 int32_t channel_dim_index,
73 int32_t channel_dim_index,
107 int32_t channel_dim_index,
|
| D | quantize_model.cc | 143 bool is_per_channel, int channel_dim_index, in QuantizeBias() argument
156 if (bias_tensor->shape[0] != weight_tensor->shape[channel_dim_index]) { in QuantizeBias()
160 bias_tensor->shape[0], weight_tensor->shape[channel_dim_index]); in QuantizeBias()
1214 int32_t channel_dim_index = input.second.per_axis_index; in FillQuantizationParams() local
1216 float_input_data, tensor->shape, channel_dim_index, in FillQuantizationParams()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1152 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormShape() local
1153 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormShape()
1163 TF_RETURN_IF_ERROR(c->ReplaceDim(x, channel_dim_index, channel_dim, &y)); in FusedBatchNormShape()
1192 int channel_dim_index = GetTensorFeatureDimIndex(4, data_format); in FusedBatchNormExShape() local
1193 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormExShape()
1222 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormGradShape() local
1223 DimensionHandle channel_dim = c->Dim(y_backprop, channel_dim_index); in FusedBatchNormGradShape()
1225 c->Merge(channel_dim, c->Dim(x, channel_dim_index), &channel_dim)); in FusedBatchNormGradShape()
1236 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()
|