Searched refs:check_axis (Results 1 – 1 of 1) sorted by relevance
1069 auto check_axis = [](int64_t &axis, const size_t dim) -> void { in InferImplArgMaxWithValue() local1080 …auto cal_shape = [axis, keep_dims_value, check_axis](ShapeVector &shape, const ShapeVector &x_shap… in InferImplArgMaxWithValue()1083 check_axis(axis_value, x_shape.size()); in InferImplArgMaxWithValue()