From 1d7457efb53e7b18fd72645d54a3404b1e782906 Mon Sep 17 00:00:00 2001 From: yanzhenxiang2020 Date: Mon, 12 Apr 2021 21:02:16 +0800 Subject: [PATCH] add FusedSparseAdam Ascend docs --- mindspore/ops/operations/nn_ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index e1932b1cba..23caca2ffe 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -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