Searched refs:HookBackward (Results 1 – 7 of 7) sorted by relevance
315 class HookBackward(PrimitiveWithInfer): class358 super(HookBackward, self).__init__(self.__class__.__name__)
42 from .debug_ops import (ImageSummary, InsertGradientOf, HookBackward, ScalarSummary,
34 self.hook = P.HookBackward(var_hook_function)48 self.hook = P.HookBackward(var_hook_function)
110 self.hook = P.HookBackward(var_hook_function)
87 self.hook = P.HookBackward(hook_function)
36 from ..ops.operations import HookBackward, Cast1297 self._backward_hook = HookBackward(fn, self.cls_name + "(" + str(id(self)) + ")")
597 self.hook = P.HookBackward(var_hook_function)