Browse Source

!6879 Fix bug of PrimitiveWithCheck doc

Merge pull request !6879 from fary86/fix_doc_of_PrimitiveWithCheck
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
0d3c257ec0
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ops/primitive.py

+ 2
- 2
mindspore/ops/primitive.py View File

@@ -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.


Loading…
Cancel
Save