Searched refs:cal_shape (Results 1 – 1 of 1) sorted by relevance
1080 …auto cal_shape = [axis, keep_dims_value, check_axis](ShapeVector &shape, const ShapeVector &x_shap… in InferImplArgMaxWithValue() local1097 cal_shape(shape, x_shape); in InferImplArgMaxWithValue()1098 cal_shape(min_shape, x_min_shape); in InferImplArgMaxWithValue()1099 cal_shape(max_shape, x_max_shape); in InferImplArgMaxWithValue()