/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
checkpoint_input_pipeline_hook_test.py | 41 def _model_fn(features, labels, mode, config): member in class:CheckpointInputPipelineHookTest 77 est = estimator.Estimator(model_fn=self._model_fn) 91 est = estimator.Estimator(model_fn=self._model_fn) 103 est = estimator.Estimator(model_fn=self._model_fn) 118 est = estimator.Estimator(model_fn=self._model_fn)
|
/external/tensorflow/tensorflow/contrib/predictor/ |
testing_common.py | 48 def _model_fn(features, labels, mode): function in function:get_arithmetic_estimator 76 return core_estimator.Estimator(_model_fn) 78 return contrib_estimator.Estimator(_model_fn, model_dir=model_dir)
|
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
estimator.py | 629 def _model_fn(features, labels, mode, config): function in function:CoreGradientBoostedDecisionTreeEstimator.__init__ 652 model_fn=_model_fn, model_dir=model_dir, config=config) 710 def _model_fn(features, labels, mode, config): function in function:CoreGradientBoostedDecisionTreeRanker.__init__ 799 def _model_fn(features, labels, mode, config): function in function:CoreGradientBoostedDecisionTreeQuantileRegressor.__init__ [all...] |
dnn_tree_combined_estimator.py | 485 def _model_fn(features, labels, mode, config): function in function:DNNBoostedTreeCombinedClassifier.__init__ 511 model_fn=_model_fn, 616 def _model_fn(features, labels, mode, config): function in function:DNNBoostedTreeCombinedRegressor.__init__ 642 model_fn=_model_fn, 732 def _model_fn(features, labels, mode, config): function in function:DNNBoostedTreeCombinedEstimator.__init__ 832 def _model_fn(features, labels, mode, config): function in function:CoreDNNBoostedTreeCombinedEstimator.__init__ [all...] |
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
logistic_regressor.py | 50 def _model_fn(features, labels, mode, params): function in function:_get_model_fn_with_logistic_metrics 74 return _model_fn
|
estimator_test.py | 342 def _model_fn(arg0, arg1, mode, params, config): function in function:EstimatorModelFnTest.testModelFnArgs 351 model_fn=_model_fn, params=expected_params, config=expected_config) 370 def _model_fn(arg0, arg1, foo, mode, params, config, bar): function in function:EstimatorModelFnTest.testPartialModelFnArgs 381 _model_fn, foo=expected_foo, bar=expected_bar) [all...] |
/external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
gmm.py | 154 def _model_fn(features, labels, mode, config): function in function:GMM._model_builder 183 return _model_fn
|
/external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
random_forest.py | 171 def _model_fn(features, labels, mode): function in function:get_model_fn 329 return _model_fn 499 def _model_fn(features, labels, mode): function in function:get_combined_model_fn 540 return _model_fn
|
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
gan_estimator_impl.py | 175 def _model_fn(features, labels, mode): function in function:GANEstimator.__init__ 196 model_fn=_model_fn, model_dir=model_dir, config=config,
|
stargan_estimator_impl.py | 158 def _model_fn(features, labels, mode): function in function:StarGANEstimator.__init__ 185 model_fn=_model_fn, model_dir=model_dir, config=config)
|
tpu_gan_estimator_impl.py | 223 def _model_fn(features, labels, mode, params): function in function:TPUGANEstimator.__init__ 250 model_fn=_model_fn,
|
/external/tensorflow/tensorflow/contrib/compiler/ |
xla.py | 564 self._model_fn = function 675 model_fn_args = function_utils.fn_args(self._model_fn) 690 self._model_fn(features=features, **kwargs))
|
/external/tensorflow/tensorflow/python/tpu/ |
tpu_estimator.py | 2769 def _model_fn(features, labels, mode, config, params): function in function:TPUEstimator._augment_model_fn [all...] |
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
keras_support.py | 946 def _model_fn(): function in function:TPUFunction._specialize_model [all...] |