Searched refs:_gradient_function (Results 1 – 3 of 3) sorted by relevance
129 def _gradient_function(op_name, attr_tuple, num_inputs, inputs, outputs, function164 pywrap_tfe.TFE_Py_RegisterGradientFunction(_gradient_function)
392 op._gradient_function = function._get_gradient_function() # pylint: disable=protected-access
2019 self._gradient_function = None2711 gradient_function = op._gradient_function # pylint: disable=protected-access3615 …op._gradient_function = self._gradient_function_map.get(op.type) # pylint: disable=protected-acce…