Searched refs:InsertGradientOf (Results 1 – 8 of 8) sorted by relevance
42 stop = P.InsertGradientOf(stop_gradient)72 clip = P.InsertGradientOf(clip_gradient)132 self.get_g = P.InsertGradientOf(self.save_gradient)
61 @bprop_getters.register(P.InsertGradientOf)
101 self.print_shape_type = P.InsertGradientOf(PrintShapeType(msg))
251 class InsertGradientOf(PrimitiveWithInfer): class
42 from .debug_ops import (ImageSummary, InsertGradientOf, HookBackward, ScalarSummary,
135 self.getG = P.InsertGradientOf(self.save_gradient)406 self.getG = P.InsertGradientOf(self.save_gradient)604 self.getG = P.InsertGradientOf(self.save_gradient)821 self.getG = P.InsertGradientOf(self.save_gradient)
116 self.getG = P.InsertGradientOf(self.save_gradient)330 self.getG = P.InsertGradientOf(self.save_gradient)
271 self.getG = P.InsertGradientOf(self.save_gradient)