Searched refs:set_hook (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | primitive_py.h | 60 void set_hook(const py::function &hook) { hook_ = hook; } in set_hook() function
100 void set_hook(const py::function &hook);
|
| D | primitive_py.cc | 71 set_hook(adapter->hook_); in PrimitivePy()
318 this->set_hook(primitive_py->hook()); in CopyHookFunction()
480 void PrimitivePyAdapter::set_hook(const py::function &hook) { in set_hook() function in mindspore::PrimitivePyAdapter
484 prim->set_hook(hook); in set_hook()
523 … .def("register_hook", &PrimitivePyAdapter::set_hook, "Set primitive hook function.") in __anonda41446c0202()
|
| /third_party/python/Modules/ |
| D | readline.c | 397 set_hook(const char *funcname, PyObject **hook_var, PyObject *function) in set_hook() function
433 PyObject *result = set_hook("completion_display_matches_hook", in readline_set_completion_display_matches_hook_impl()
467 return set_hook("startup_hook", &readlinestate_global->startup_hook, in readline_set_startup_hook_impl()
492 return set_hook("pre_input_hook", &readlinestate_global->pre_input_hook, in readline_set_pre_input_hook_impl()
769 return set_hook("completer", &readlinestate_global->completer, function); in readline_set_completer_impl()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | data_converter.cc | 128 fake_bprop->set_hook(bprop_func); in ConvertToBpropCut()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 2470 fake_prim->set_hook(bprop_func); in DoGradForCustomBprop()
|