Searched refs:_add_scope_exit_callback (Results 1 – 2 of 2) sorted by relevance
801 def _add_scope_exit_callback(self, fn): member in FuncGraph
6917 default_graph._add_scope_exit_callback(fn) # pylint: disable=protected-access