modify annotations for inputs of sparse adam

pull/2158/head
wangnan39@huawei.com 5 years ago
parent f48ba77648
commit 9fe3856b9f

@ -2777,7 +2777,7 @@ class SparseApplyAdam(PrimitiveWithInfer):
- **m** (Parameter) - The 1st moment vector in the updating formula. Has the same type as `var`. With
float32 data type.
- **v** (Parameter) - The 2nd moment vector in the updating formula. Mean square gradients,
has the same type as `var`. With float32 data type.
has the same type as `var`. With float32 data type.
- **beta1_power** (Tensor) - :math:`beta_1^t` in the updating formula. With float32 data type.
- **beta2_power** (Tensor) - :math:`beta_2^t` in the updating formula. With float32 data type.
- **lr** (Tensor) - :math:`l` in the updating formula. With float32 data type.
@ -2911,7 +2911,7 @@ class SparseApplyLazyAdam(PrimitiveWithInfer):
- **m** (Parameter) - The 1st moment vector in the updating formula. Has the same type as `var`. With
float32 data type.
- **v** (Parameter) - The 2nd moment vector in the updating formula. Mean square gradients,
has the same type as `var`. With float32 data type.
has the same type as `var`. With float32 data type.
- **beta1_power** (Tensor) - :math:`beta_1^t` in the updating formula. With float32 data type.
- **beta2_power** (Tensor) - :math:`beta_2^t` in the updating formula. With float32 data type.
- **lr** (Tensor) - :math:`l` in the updating formula. With float32 data type.

Loading…
Cancel
Save