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

!2355 fix cpu sub op input not match 

Merge pull request !2355 from dengwentao/fix_sub_op
This commit is contained in:
mindspore-ci-bot 2020-06-20 11:25:49 +08:00 committed by Gitee
parent b8a21ac6e8 92744f1301
commit bdb7d0fd01
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/kernel/cpu/sub_cpu_kernel.h
View File

@ -36,7 +36,9 @@ class SubCPUKernel : public CPUKernel {
int offset_;
};
MS_REG_CPU_KERNEL(Sub, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt32), SubCPUKernel);
MS_REG_CPU_KERNEL(
Sub, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt32),
SubCPUKernel);
} // namespace kernel
} // namespace mindspore