Home
last modified time | relevance | path

Searched refs:kPrimSparseTensorGetDenseShape (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dsparse_tensor_eliminate.h41 MATCH_REPLACE(node, PPrimitive(prim::kPrimSparseTensorGetDenseShape, sparse), z); in operator()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dresource.cc241 {"dense_shape", prim::kPrimSparseTensorGetDenseShape}, // F.sparse_tensor_get_dense_shape in GetAttrMap()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dirpass.cc247 …imSparseTensorGetIndices, prim::kPrimSparseTensorGetValues, prim::kPrimSparseTensorGetDenseShape}); in OptimizeIRPassLib()
Dclean.cc503 } else if (IsPrimitiveCNode(node, prim::kPrimSparseTensorGetDenseShape) || in CleanAfterOptA()
/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.cc192 {prim::kPrimSparseTensorGetDenseShape, {InferImplSparseTensorGetDenseShape, nullptr, true}}, in GetPrimitiveToEvalImplMap()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h454 inline const PrimitivePtr kPrimSparseTensorGetDenseShape = std::make_shared<Primitive>("SparseTenso… variable