Searched refs:_gradient_function (Results 1 – 3 of 3) sorted by relevance
127 def _gradient_function(op_name, attr_tuple, num_inputs, inputs, outputs, function162 pywrap_tfe.TFE_Py_RegisterGradientFunction(_gradient_function)
486 op._gradient_function = function._get_gradient_function() # pylint: disable=protected-access
2170 self._gradient_function = None2874 gradient_function = op._gradient_function # pylint: disable=protected-access3868 …op._gradient_function = self._gradient_function_map.get(op.type) # pylint: disable=protected-acce…