Searched refs:with_mode (Results 1 – 2 of 2) sorted by relevance
2488 with self._ctx.with_mode(mode) as ctx:2588 with self._ctx.with_mode(model_fn_lib.ModeKeys.TRAIN) as ctx:2610 with self._ctx.with_mode(model_fn_lib.ModeKeys.EVAL) as ctx:2658 with self._ctx.with_mode(mode) as ctx:2789 with self._ctx.with_mode(mode) as ctx:
236 def with_mode(self, mode): member in _InternalTPUContext