Searched refs:hook_grad_ (Results 1 – 2 of 2) sorted by relevance
59 std::map<std::string, py::object> PrimitivePy::hook_grad_; member in mindspore::PrimitivePy153 hook_grad_.clear(); in CheckHookConsistency()159 hook_grad_.clear(); in CheckHookConsistency()170 hook_grad_.clear(); in CheckHookConsistency()178 hook_grad_.clear(); in CheckHookConsistency()220 auto iter = hook_grad_.find(cell_id); in RunCellHookFunction()221 if (iter != hook_grad_.end()) { in RunCellHookFunction()242 (void)hook_grad_.erase(cell_id); in RunCellHookFunction()244 hook_grad_[cell_id] = py_args[grad_output_index]; in RunCellHookFunction()
86 static std::map<std::string, py::object> hook_grad_; variable