Searched defs:TensorBackwardHook (Results 1 – 3 of 3) sorted by relevance
35 struct COMMON_EXPORT TensorBackwardHook : public BackWardHook { struct39 std::map<uint64_t, py::function> hook_map_;
75 TensorBackwardHook::TensorBackwardHook(uint64_t tensor_id, const py::function &obj) { in TensorBackwardHook() function in mindspore::TensorBackwardHook
33 class TensorBackwardHook; variable