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

check if primtive is null

This commit is contained in:
VectorSL 2022-04-28 15:15:20 +08:00
parent aca75c0200
commit 5c071df2a6
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/core/ops/sparse_matrix_add.cc
View File

@ -67,6 +67,7 @@ void SparseMatrixAdd::Init(const std::vector<int64_t> &csr_a, const std::vector<
AbstractBasePtr SparseMatrixAddInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<AbstractBasePtr> &input_args) {
MS_EXCEPTION_IF_NULL(primitive);
const std::string op_name = primitive->name();
// Addition of two CSR tensors. C = Alpha * A + Beta * B.
// Eight input (CSR_A(three tensors), CSR_B(three tensors), Alpha, Beta)