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