Searched refs:_call_model_fn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla.py | 578 return self._call_model_fn(features, labels, mode, params)
618 estimator_spec = self._call_model_fn(features, labels,
643 estimator_spec = self._call_model_fn(features, labels,
673 def _call_model_fn(self, features, labels, mode, params): member in _ModelFnWrapper
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | estimator.py | 483 return self._call_model_fn(features, labels, mode, config=config)
1196 def _call_model_fn(self, features, labels, mode, metrics=None, config=None): member in Estimator
1264 return self._call_model_fn(features, labels, model_fn_lib.ModeKeys.TRAIN)
1291 model_fn_ops = self._call_model_fn(features, labels,
1314 return self._call_model_fn(features, labels, model_fn_lib.ModeKeys.INFER)
1398 model_fn_ops = self._call_model_fn(features, None,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 1412 return self._call_model_fn(features, labels, is_export_mode=is_export_mode)
1463 self._call_model_fn(features, labels))
1535 tpu_estimator_spec = self._call_model_fn(features, labels)
1583 tpu_estimator_spec = self._call_model_fn(
1661 def _call_model_fn(self, features, labels, is_export_mode=False): member in _ModelFnWrapper
2426 def _call_model_fn(self, features, labels, mode, config): member in TPUEstimator
2430 return super(TPUEstimator, self)._call_model_fn(features, labels, mode,
2543 estimator_spec = super(TPUEstimator, self)._call_model_fn(
|