Searched refs:_GetModelFnOpsForMode (Results 1 – 1 of 1) sorted by relevance
225 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.TRAIN)233 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.EVAL)241 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.INFER)248 def _GetModelFnOpsForMode(self, mode): member in DynamicRnnEstimatorTest