Searched defs:channel_dim_index (Results 1 – 3 of 3) sorted by relevance
136 int32_t channel_dim_index, in FillPerChannelMinMax()293 int32_t channel_dim_index, in SymmetricPerChannelQuantization()373 int32_t channel_dim_index, in SymmetricPerChannelQuantizeValues()560 int32_t channel_dim_index, in SymmetricQuantizeTensorPerChannel()
143 bool is_per_channel, int channel_dim_index, in QuantizeBias()1214 int32_t channel_dim_index = input.second.per_axis_index; in FillQuantizationParams() local
1152 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormShape() local1192 int channel_dim_index = GetTensorFeatureDimIndex(4, data_format); in FusedBatchNormExShape() local1222 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormGradShape() local