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