Home
last modified time | relevance | path

Searched refs:channel_dim_index (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils.cc83 int32_t channel_dim_index, in SymmetricPerChannelQuantization() argument
86 const int32_t channel_dim_size = dimension[channel_dim_index]; in SymmetricPerChannelQuantization()
99 int channel_idx = indices[channel_dim_index]; in SymmetricPerChannelQuantization()
133 channel_dim_index, output_value); in SymmetricPerChannelQuantization()
139 int32_t channel_dim_index, in SymmetricPerChannelQuantizeValues() argument
149 int channel_idx = indices[channel_dim_index]; in SymmetricPerChannelQuantizeValues()
Dsubgraph_quantizer.cc89 int32_t channel_dim_index, in SymmetricPerChannelQuantizeTensor() argument
100 const int32_t channel_dim_size = tensor->shape[channel_dim_index]; in SymmetricPerChannelQuantizeTensor()
114 float_input_data, tensor_dims, channel_dim_index, &scales, &final_buffer); in SymmetricPerChannelQuantizeTensor()
120 return AddQuantizationParams(scales, zero_point, channel_dim_index, in SymmetricPerChannelQuantizeTensor()
129 int channel_dim_index, in SymmetricPerChannelBiasQuantize() argument
142 if (tensor->shape[0] != weight_tensor->shape[channel_dim_index]) { in SymmetricPerChannelBiasQuantize()
145 tensor->shape[0], weight_tensor->shape[channel_dim_index]); in SymmetricPerChannelBiasQuantize()
190 return AddQuantizationParams(scales, zero_point, channel_dim_index, in SymmetricPerChannelBiasQuantize()
Dquantization_utils.h52 int32_t channel_dim_index,
60 int32_t channel_dim_index,
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc803 int channel_dim_index = GetTensorFeatureDimIndex(4, data_format); in FusedBatchNormShape() local
804 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormShape()
814 TF_RETURN_IF_ERROR(c->ReplaceDim(x, channel_dim_index, channel_dim, &y)); in FusedBatchNormShape()
839 int channel_dim_index = GetTensorFeatureDimIndex(4, data_format); in FusedBatchNormGradShape() local
840 DimensionHandle channel_dim = c->Dim(y_backprop, channel_dim_index); in FusedBatchNormGradShape()
842 c->Merge(channel_dim, c->Dim(x, channel_dim_index), &channel_dim)); in FusedBatchNormGradShape()
853 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()