Searched defs:num_elements_in_axis (Results 1 – 3 of 3) sorted by relevance
42 const int num_elements_in_axis = input_width * input_height; in Mean() local
3584 U num_elements_in_axis = 1; in Mean() local3669 const float num_elements_in_axis = input_width * input_height; in Mean() local3750 U num_elements_in_axis = 1; in QuantizedMeanOrSum() local
1771 const float num_elements_in_axis = input_width * input_height; in MeanImpl() local