Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!49767 adam check dtype 

Merge pull request !49767 from nomindcarry/master
This commit is contained in:
i-robot 2023-03-06 06:16:20 +00:00 committed by Gitee
parent b528aac206 f89ac63375
commit 93e13a453b
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mindspore/python/mindspore/ops/operations/nn_ops.py
View File

@ -4634,14 +4634,14 @@ class Adam(Primitive):
"""
__mindspore_signature__ = (
sig.make_sig('var', sig.sig_rw.RW_WRITE, dtype=sig.sig_dtype.T),
sig.make_sig('m', sig.sig_rw.RW_WRITE, dtype=sig.sig_dtype.T),
sig.make_sig('v', sig.sig_rw.RW_WRITE, dtype=sig.sig_dtype.T),
sig.make_sig('beta1_power', dtype=sig.sig_dtype.T),
sig.make_sig('beta2_power', dtype=sig.sig_dtype.T),
sig.make_sig('lr', dtype=sig.sig_dtype.T),
sig.make_sig('beta1', dtype=sig.sig_dtype.T),
sig.make_sig('beta2', dtype=sig.sig_dtype.T),
sig.make_sig('epsilon', dtype=sig.sig_dtype.T1),
sig.make_sig('m', sig.sig_rw.RW_WRITE, dtype=sig.sig_dtype.T1),
sig.make_sig('v', sig.sig_rw.RW_WRITE, dtype=sig.sig_dtype.T2),
sig.make_sig('beta1_power', dtype=sig.sig_dtype.T3),
sig.make_sig('beta2_power', dtype=sig.sig_dtype.T4),
sig.make_sig('lr', dtype=sig.sig_dtype.T5),
sig.make_sig('beta1', dtype=sig.sig_dtype.T6),
sig.make_sig('beta2', dtype=sig.sig_dtype.T7),
sig.make_sig('epsilon', dtype=sig.sig_dtype.T8),
sig.make_sig('gradient', dtype=sig.sig_dtype.T)
)