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

fix sub input not match

This commit is contained in:
dengwentao 2020-06-19 16:32:02 +08:00
parent 2ad211df97
commit 92744f1301
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