Searched refs:kPrimRowTensorGetDenseShape (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | row_tensor_eliminate.h | 41 MATCH_REPLACE(node, PPrimitive(prim::kPrimRowTensorGetDenseShape, slices), z); in operator()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.cc | 235 {"dense_shape", prim::kPrimRowTensorGetDenseShape}, // F.row_tensor_get_dense_shape in GetAttrMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 238 …prim::kPrimRowTensorGetIndices, prim::kPrimRowTensorGetValues, prim::kPrimRowTensorGetDenseShape}); in OptimizeIRPassLib()
|
| D | clean.cc | 504 IsPrimitiveCNode(node, prim::kPrimRowTensorGetDenseShape)) { in CleanAfterOptA()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 198 {prim::kPrimRowTensorGetDenseShape, {InferImplRowTensorGetDenseShape, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 447 inline const PrimitivePtr kPrimRowTensorGetDenseShape = std::make_shared<Primitive>("RowTensorGetDe… variable
|