Searched refs:num_axis_dimensions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3523 const int* axis, const int64_t num_axis_dimensions, in ReduceGeneric() argument
3535 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in ReduceGeneric()
3552 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean() argument
3572 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in Mean()
3716 const int num_axis_dimensions, bool keep_dims, in QuantizedMeanOrSum() argument
3738 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in QuantizedMeanOrSum()
|