Searched defs:num_axis_dimensions (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | reduce.h | 407 const int* axis, const int num_axis_dimensions, in QuantizedMeanOrSum()
498 const int* axis, const int64_t num_axis_dimensions, in ReduceDispatcher()
624 const int64_t num_axis_dimensions, int* resolved_axis, int* normalized_dims, in QuantizedReduceProd()
674 const int* axis, const int num_axis_dimensions, in MeanGeneral()
732 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean()
782 const int* axis, const int64_t num_axis_dimensions, in ReduceGeneric()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reduce.h | 183 const int* axis, const int64_t num_axis_dimensions, in ReduceGeneric()
220 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean()
384 const int num_axis_dimensions, bool keep_dims, in QuantizedMeanOrSum()
477 const int64_t num_axis_dimensions, in QuantizedReduceProd()
|