| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | checkpoint_input_pipeline_hook_test.py | 41 def _model_fn(features, labels, mode, config): member in 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/boosted_trees/estimator_batch/ |
| D | estimator.py | 629 def _model_fn(features, labels, mode, config): function
652 model_fn=_model_fn, model_dir=model_dir, config=config)
710 def _model_fn(features, labels, mode, config): function
735 model_fn=_model_fn, model_dir=model_dir, config=config)
799 def _model_fn(features, labels, mode, config): function
837 model_fn=_model_fn, model_dir=model_dir, config=config)
|
| D | dnn_tree_combined_estimator.py | 485 def _model_fn(features, labels, mode, config): function
511 model_fn=_model_fn,
616 def _model_fn(features, labels, mode, config): function
642 model_fn=_model_fn,
732 def _model_fn(features, labels, mode, config): function
758 model_fn=_model_fn,
832 def _model_fn(features, labels, mode, config): function
859 model_fn=_model_fn, model_dir=model_dir, config=config)
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| D | testing_common.py | 48 def _model_fn(features, labels, mode): function
76 return core_estimator.Estimator(_model_fn)
78 return contrib_estimator.Estimator(_model_fn, model_dir=model_dir)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | logistic_regressor.py | 50 def _model_fn(features, labels, mode, params): function
74 return _model_fn
|
| D | estimator.py | 1191 self._model_fn = model_fn
1214 model_fn_args = _model_fn_args(self._model_fn)
1227 model_fn_results = self._model_fn(features, labels, **kwargs)
|
| D | estimator_test.py | 342 def _model_fn(arg0, arg1, mode, params, config): function
351 model_fn=_model_fn, params=expected_params, config=expected_config)
370 def _model_fn(arg0, arg1, foo, mode, params, config, bar): function
381 _model_fn, foo=expected_foo, bar=expected_bar)
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | gmm.py | 154 def _model_fn(features, labels, mode, config): function
183 return _model_fn
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| D | random_forest.py | 171 def _model_fn(features, labels, mode): function
329 return _model_fn
499 def _model_fn(features, labels, mode): function
540 return _model_fn
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | gan_estimator_impl.py | 175 def _model_fn(features, labels, mode): function
196 model_fn=_model_fn, model_dir=model_dir, config=config,
|
| D | stargan_estimator_impl.py | 158 def _model_fn(features, labels, mode): function
185 model_fn=_model_fn, model_dir=model_dir, config=config)
|
| D | tpu_gan_estimator_impl.py | 223 def _model_fn(features, labels, mode, params): function
250 model_fn=_model_fn,
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | 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/ |
| D | tpu_estimator.py | 1406 self._model_fn = model_fn
1664 model_fn_args = function_utils.fn_args(self._model_fn)
1686 'params[\'batch_size\']'.format(self._model_fn))
1707 estimator_spec = self._model_fn(features=features, **kwargs)
2769 def _model_fn(features, labels, mode, config, params): function
3105 return _model_fn
3792 estimator._model_fn, # pylint: disable=protected-access
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 946 def _model_fn(): function
1096 _model_fn, inputs=[[] for _ in range(self._tpu_assignment.num_towers)])
|