fix sub input not match

92744f13 · dengwentao · 2ad211df · 92744f13
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

mindspore/ccsrc/kernel/cpu/sub_cpu_kernel.h mindspore/ccsrc/kernel/cpu/sub_cpu_kernel.h +3 -1

未找到文件。
--- a/mindspore/ccsrc/kernel/cpu/sub_cpu_kernel.h
+++ b/mindspore/ccsrc/kernel/cpu/sub_cpu_kernel.h
@@ -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