Browse Source

!15027 add FusedSparseAdam Ascend docs

From: @yanzhenxiang2020
Reviewed-by: @oacjiewen,@wuxuejian
Signed-off-by: @wuxuejian
tags/v1.2.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
8b19b69fc7
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ops/operations/nn_ops.py

+ 2
- 2
mindspore/ops/operations/nn_ops.py View File

@@ -4410,7 +4410,7 @@ class FusedSparseAdam(PrimitiveWithInfer):
`gradient` or `indices` is not float32.

Supported Platforms:
``CPU``
``Ascend`` ``CPU``

Examples:
>>> import numpy as np
@@ -4558,7 +4558,7 @@ class FusedSparseLazyAdam(PrimitiveWithInfer):
TypeError: If dtype of `indices` is not int32.

Supported Platforms:
``CPU``
``Ascend`` ``CPU``

Examples:
>>> import numpy as np


Loading…
Cancel
Save