| /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...] |
