Searched refs:_gradient_registry (Results 1 – 5 of 5) sorted by relevance
310 for op_type in ops._gradient_registry.list(): # pylint: disable=protected-access315 gradient_fn = ops._gradient_registry.lookup(op_type) # pylint: disable=protected-access
148 grad_fn = ops._gradient_registry.lookup(op_name) # pylint: disable=protected-access
119 ops._gradient_registry.lookup(op_name) # pylint: disable=protected-access
27 …return ops._gradient_registry.lookup(op.op_def.name) is not None # pylint: disable=protected-acce…
2615 _gradient_registry = registry.Registry("gradient") variable2662 _gradient_registry.register(f, self._op_type)2698 _gradient_registry.register(None, op_type)2719 return _gradient_registry.lookup(op_type)