wangtao
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add an output to apply_proximal_adagrad op register

This commit is contained in:
yujianfeng 2020-06-23 11:54:51 +08:00
parent 35ab95bfae
commit 34407391e6
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/kernel/cpu/sparse_apply_proximal_adagrad_cpu_kernel.h
View File

@ -48,6 +48,7 @@ MS_REG_CPU_KERNEL(SparseApplyProximalAdagrad,
.AddInputAttr(kNumberTypeFloat32)
.AddInputAttr(kNumberTypeFloat32)
.AddInputAttr(kNumberTypeInt32)
.AddOutputAttr(kNumberTypeFloat32)
.AddOutputAttr(kNumberTypeFloat32),
SparseApplyProximalAdagradCPUKernel);
} // namespace kernel

2
tests/st/ops/cpu/test_sparse_apply_proximal_adagrad_op.py
View File

@ -44,4 +44,4 @@ def test_net():
context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
sparse_apply_proximal_adagrad = Net()
output = sparse_apply_proximal_adagrad(gradient, indices)
print(output.asnumpy()[0])
print(output[0].asnumpy())