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

!49831 Add shape type registration 

Merge pull request !49831 from NaCN/fix_shape
This commit is contained in:
i-robot 2023-03-06 11:32:56 +00:00 committed by Gitee
parent 1dba3968fa daca853297
commit a89ba13372
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mindspore/ccsrc/plugin/device/cpu/kernel/shape_cpu_kernel.cc
View File

@ -70,14 +70,29 @@ const std::vector<std::pair<KernelAttr, ShapeCpuKernelMod::KernelRunFunc>> &Shap
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel}};
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeComplex64).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
{KernelAttr().AddInputAttr(kNumberTypeComplex128).AddOutputAttr(kObjectTypeTuple, kNumberTypeInt64),
&ShapeCpuKernelMod::LaunchKernel},
};
return func_list;
}