Home
last modified time | relevance | path

Searched defs:num_elements_in_axis (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreduce.h252 size_t num_elements_in_axis = 1; in Mean() local
337 const float num_elements_in_axis = input_width * input_height; in Mean() local
433 size_t num_elements_in_axis = 1; in QuantizedMeanOrSum() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dreduce.h220 const float num_elements_in_axis = input_width * input_height; in Mean() local
455 size_t num_elements_in_axis = 1; in QuantizedMeanOrSum() local
709 size_t num_elements_in_axis = 1; in MeanGeneral() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dmean.h45 const int num_elements_in_axis = input_width * input_height; in Mean() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dmean.h207 const float num_elements_in_axis = input_width * input_height; in Mean() local