Searched refs:_maybe_wrap_with_training_arg (Results 1 – 1 of 1) sorted by relevance
456 def _maybe_wrap_with_training_arg(self, call_fn): member in LayerCallCollection497 self, self._maybe_wrap_with_training_arg(call_fn), name,