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
197 size_t num_elements_in_axis = 1; in Mean() local282 const float num_elements_in_axis = input_width * input_height; in Mean() local370 size_t num_elements_in_axis = 1; in QuantizedMeanOrSum() local
205 const float num_elements_in_axis = input_width * input_height; in Mean() local
1155 const float num_elements_in_axis = input_width * input_height; in Mean() local