From 3abf516fd4cae4743bcd9fa3b3465e26e231a315 Mon Sep 17 00:00:00 2001 From: fary86 Date: Fri, 25 Sep 2020 11:36:56 +0800 Subject: [PATCH] Fix bug of PrimitiveWithCheck doc --- mindspore/ops/primitive.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ops/primitive.py b/mindspore/ops/primitive.py index 25e05749f6..3b217719e2 100644 --- a/mindspore/ops/primitive.py +++ b/mindspore/ops/primitive.py @@ -200,8 +200,8 @@ class PrimitiveWithCheck(Primitive): There are three methods can be overide to define the check logic of the primitive: __check__(), check_shape(), check_dtype(). If __check__() is defined in primitive, the __check__() has highest priority to be called. - If __check__() is not defined, infer_shape() and infer_dtype() can be defined to describe the check logic of - the shape and type. + If __check__() is not defined, check_shape() and check_dtype() can be defined to describe the check logic of + the shape and type. Method infer_value() can also be defined (such as PrimitiveWithInfer) for constant propagation. Args: name (str): Name of the current Primitive.