Searched refs:_register_gradient (Results 1 – 3 of 3) sorted by relevance
116 output_func._register_gradient() # pylint: disable=protected-access
611 self._register_gradient()618 def _register_gradient(self): member in ConcreteFunction829 self._register_gradient()876 self._register_gradient()
256 func._register_gradient() # pylint: disable=protected-access