Searched defs:num_axis_dimensions (Results 1 – 1 of 1) sorted by relevance
3523 const int* axis, const int64_t num_axis_dimensions, in ReduceGeneric()3552 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean()3716 const int num_axis_dimensions, bool keep_dims, in QuantizedMeanOrSum()