Searched refs:_gradient_registry (Results 1 – 2 of 2) sorted by relevance
126 grad_fn = ops._gradient_registry.lookup(op_name) # pylint: disable=protected-access
2278 _gradient_registry = registry.Registry("gradient") variable2320 _gradient_registry.register(f, self._op_type)2355 _gradient_registry.register(None, op_type)2370 return _gradient_registry.lookup(op_type)