Searched defs:num_axis_dimensions (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 181 const int* axis, const int64_t num_axis_dimensions, in ReduceGeneric()
218 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean()
382 const int num_axis_dimensions, bool keep_dims, in QuantizedMeanOrSum()
475 const int64_t num_axis_dimensions, in QuantizedReduceProd()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1198 const int* axis, const int num_axis_dimensions, in MeanGeneral()
|