Searched defs:num_axis_dimensions (Results 1 – 2 of 2) sorted by relevance
131 const int* axis, const int64_t num_axis_dimensions, in ReduceGeneric()165 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean()329 const int num_axis_dimensions, bool keep_dims, in QuantizedMeanOrSum()
1200 const int* axis, const int num_axis_dimensions, in MeanGeneral()