Searched defs:num_elements_in_axis (Results 1 – 4 of 4) sorted by relevance
43 const int num_elements_in_axis = input_width * input_height; in Mean() local
250 size_t num_elements_in_axis = 1; in Mean() local335 const float num_elements_in_axis = input_width * input_height; in Mean() local431 size_t num_elements_in_axis = 1; in QuantizedMeanOrSum() local
205 const float num_elements_in_axis = input_width * input_height; in Mean() local
1153 const float num_elements_in_axis = input_width * input_height; in Mean() local