Searched refs:channel_dim_index (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/tools/optimize/ |
D | quantization_utils.cc | 83 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()
|
D | subgraph_quantizer.cc | 89 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()
|
D | quantization_utils.h | 52 int32_t channel_dim_index, 60 int32_t channel_dim_index,
|
/external/tensorflow/tensorflow/core/framework/ |
D | common_shape_fns.cc | 803 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()
|