Browse Source

!8568 add int32 kernel reg for sparse to dense

From: @zhaozhenlong
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
9c8828761d
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/fp32/sparse_to_dense_fp32.cc

+ 1
- 0
mindspore/lite/src/runtime/kernel/arm/fp32/sparse_to_dense_fp32.cc View File

@@ -211,4 +211,5 @@ kernel::LiteKernel *CpuSparseToDenseFp32KernelCreator(const std::vector<lite::Te
}

REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_SparseToDense, CpuSparseToDenseFp32KernelCreator)
REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_SparseToDense, CpuSparseToDenseFp32KernelCreator)
} // namespace mindspore::kernel

Loading…
Cancel
Save