Fix the format of note in the recompute api

pull/13373/head
yujianfeng 4 years ago
parent 7a87cae001
commit d1d728ad1c

@ -1151,10 +1151,11 @@ class Cell(Cell_):
intermediate activation computed in forward pass, we will recompute it in backward pass.
Note:
- If the computation involves something like randomization or global variable, the equivalence
is not guaranteed currently.
is not guaranteed currently.
- If the recompute api of a primitive in this cell is also called, the recompute mode of this
primitive is subject to the recompute api of the primitive.
primitive is subject to the recompute api of the primitive.
Args:
mode (bool): Specifies whether the cell is recomputed. Default: True.

@ -227,8 +227,9 @@ class Primitive(Primitive_):
pass, we will recompute it in backward pass.
Note:
- If the computation involves something like randomization or global variable, the equivalence
is not guaranteed currently.
is not guaranteed currently.
Args:
mode (bool): Specifies whether the primitive is recomputed. Default: True.

Loading…
Cancel
Save