mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!46313 [Assistant][ops] ScatterNdAdd、ScatterNdSub 

Merge pull request !46313 from Joseph_Lee/ScatterNdAdd
This commit is contained in:
i-robot 2022-12-01 11:32:05 +00:00 committed by Gitee
parent 00c176d9d2 0a2269f01e
commit 5f8c2d3f8c
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/plugin/device/cpu/kernel/scatter_nd_arithmetic_cpu_kernel.cc
View File

@ -31,7 +31,7 @@ namespace {
constexpr size_t kMinIndiceRank = 2;
template <typename T>
inline T RealDiv(const T &a, const T &b) {
constexpr T zero = T(0);
T zero = static_cast<T>(0);
if (b == zero) {
if (a == zero) {
return std::numeric_limits<T>::quiet_NaN();
@ -205,6 +205,8 @@ const ScatterNdArithmeticCpuKernelMod::ScatterNdSupportListType &ScatterNdArithm
double, int64_t)},
{SCATTER_ND_ARITHMETIC_CPU_REGISTER(kNumberTypeFloat32, kNumberTypeInt64, kNumberTypeFloat32, kNumberTypeFloat32,
float, int64_t)},
{SCATTER_ND_ARITHMETIC_CPU_REGISTER(kNumberTypeFloat16, kNumberTypeInt64, kNumberTypeFloat16, kNumberTypeFloat16,
float16, int64_t)},
{SCATTER_ND_ARITHMETIC_CPU_REGISTER(kNumberTypeInt64, kNumberTypeInt64, kNumberTypeInt64, kNumberTypeInt64, int64_t,
int64_t)},
{SCATTER_ND_ARITHMETIC_CPU_REGISTER(kNumberTypeInt32, kNumberTypeInt64, kNumberTypeInt32, kNumberTypeInt32, int32_t,
@ -225,6 +227,8 @@ const ScatterNdArithmeticCpuKernelMod::ScatterNdSupportListType &ScatterNdArithm
double, int32_t)},
{SCATTER_ND_ARITHMETIC_CPU_REGISTER(kNumberTypeFloat32, kNumberTypeInt32, kNumberTypeFloat32, kNumberTypeFloat32,
float, int32_t)},
{SCATTER_ND_ARITHMETIC_CPU_REGISTER(kNumberTypeFloat16, kNumberTypeInt32, kNumberTypeFloat16, kNumberTypeFloat16,
float16, int32_t)},
{SCATTER_ND_ARITHMETIC_CPU_REGISTER(kNumberTypeInt64, kNumberTypeInt32, kNumberTypeInt64, kNumberTypeInt64, int64_t,
int32_t)},
{SCATTER_ND_ARITHMETIC_CPU_REGISTER(kNumberTypeInt32, kNumberTypeInt32, kNumberTypeInt32, kNumberTypeInt32, int32_t,