Searched defs:arg_handler (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | convert_input_and_attr.cc | 115 ArgHandlerFunc GetArgHandlerFunc(const std::string &arg_handler) { in GetArgHandlerFunc()
127 ArgHandlerFunc GetOppArgHandlerFunc(const std::string &arg_handler) { in GetOppArgHandlerFunc()
141 … const std::string &arg_name, const std::string &arg_handler, in AddConstInputToAttr()
204 const auto &arg_handler = iter->arg_handler_; in Process() local
258 … const std::string &arg_handler, const PrimitivePtr &primitive, in AddAttrToInput()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops_generate/ |
| D | gen_ops.py | 523 def _generate_arg_handler(class_name, arg, arg_handler, is_optional): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/ |
| D | primitive.py | 100 def _set_prim_arg_with_handler(self, name, value, arg_handler): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/ |
| D | infer.cc | 332 ArgHandlerFunc GetOppArgHandlerFunc(const std::string &arg_handler) { in GetOppArgHandlerFunc()
|