|
|
|
@@ -49,6 +49,8 @@ MS_REG_CPU_KERNEL(Neg, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAt |
|
|
|
ArithmeticSelfCPUKernel); |
|
|
|
MS_REG_CPU_KERNEL(Neg, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt32), |
|
|
|
ArithmeticSelfCPUKernel); |
|
|
|
MS_REG_CPU_KERNEL(Neg, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeInt64), |
|
|
|
ArithmeticSelfCPUKernel); |
|
|
|
MS_REG_CPU_KERNEL(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32), |
|
|
|
ArithmeticSelfCPUKernel); |
|
|
|
MS_REG_CPU_KERNEL(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt32), |
|
|
|
|