Searched refs:_get_gradient_function (Results 1 – 2 of 2) sorted by relevance
392 op._gradient_function = function._get_gradient_function() # pylint: disable=protected-access
1954 {"PartitionedCall": self._get_gradient_function(),1955 "StatefulPartitionedCall": self._get_gradient_function()}):2085 def _get_gradient_function(self): member in ConcreteFunction