Searched refs:kPrimExtractKeywordArg (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph_extends.cc | 154 …specialized_graph->NewCNode({NewValueNode(prim::kPrimExtractKeywordArg), NewValueNode(kw_param_nam… in GenerateKwParams()
166 {NewValueNode(prim::kPrimExtractKeywordArg), NewValueNode(kw_param_name), param_node}); in GenerateKwParams()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 142 {prim::kPrimExtractKeywordArg, {InferImplExtractKwarg, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | clean.cc | 417 } else if (IsPrimitiveCNode(node, prim::kPrimExtractKeywordArg)) { in SimplifyDataStructures()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 629 inline const PrimitivePtr kPrimExtractKeywordArg = std::make_shared<Primitive>("extract_keyword_arg… variable
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.cc | 249 …func_graph->NewCNode({NewValueNode(prim::kPrimExtractKeywordArg), NewValueNode(kwarg_key), source_… in MixedPrecisionCastHelper()
|