Searched refs:kPrimRowTensorGetValues (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | row_tensor_eliminate.h | 40 MATCH_REPLACE(node, PPrimitive(prim::kPrimRowTensorGetValues, slices), y); in operator()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.cc | 233 {"values", prim::kPrimRowTensorGetValues}, // F.row_tensor_get_values in GetAttrMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 238 …{prim::kPrimRowTensorGetIndices, prim::kPrimRowTensorGetValues, prim::kPrimRowTensorGetDenseShape}… in OptimizeIRPassLib()
|
| D | clean.cc | 500 IsPrimitiveCNode(node, prim::kPrimRowTensorGetValues)) { in CleanAfterOptA()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 196 {prim::kPrimRowTensorGetValues, {InferImplRowTensorGetValues, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 445 inline const PrimitivePtr kPrimRowTensorGetValues = std::make_shared<Primitive>("RowTensorGetValues… variable
|