Searched refs:kPrimSparseTensorGetDenseShape (Results 1 – 6 of 6) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
D | sparse_tensor_eliminate.h | 41 MATCH_REPLACE(node, PPrimitive(prim::kPrimSparseTensorGetDenseShape, sparse), z); in operator()
|
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
D | resource.cc | 241 {"dense_shape", prim::kPrimSparseTensorGetDenseShape}, // F.sparse_tensor_get_dense_shape in GetAttrMap()
|
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
D | irpass.cc | 247 …imSparseTensorGetIndices, prim::kPrimSparseTensorGetValues, prim::kPrimSparseTensorGetDenseShape}); in OptimizeIRPassLib()
|
D | clean.cc | 503 } else if (IsPrimitiveCNode(node, prim::kPrimSparseTensorGetDenseShape) || in CleanAfterOptA()
|
/third_party/mindspore/mindspore/core/abstract/ |
D | primitive_infer_map.cc | 192 {prim::kPrimSparseTensorGetDenseShape, {InferImplSparseTensorGetDenseShape, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
/third_party/mindspore/mindspore/core/base/ |
D | core_ops.h | 454 inline const PrimitivePtr kPrimSparseTensorGetDenseShape = std::make_shared<Primitive>("SparseTenso… variable
|