Searched refs:stats_offset (Results 1 – 1 of 1) sorted by relevance
186 int stats_offset = GetOffset(input->dims->data, input->dims->size, in FlexibleLayerNorm() local189 sum_x[stats_offset] += input_val; in FlexibleLayerNorm()190 sum_xx[stats_offset] += input_val * input_val; in FlexibleLayerNorm()219 int stats_offset = GetOffset(input->dims->data, input->dims->size, in FlexibleLayerNorm() local224 input_val * multiplier[stats_offset] + bias[stats_offset]; in FlexibleLayerNorm()