Searched refs:adapt_fn (Results 1 – 1 of 1) sorted by relevance
148 adapt_fn = adapt_step151 def adapt_fn(iterator): function156 adapt_fn = def_function.function(adapt_fn)158 self._adapt_function = adapt_fn