/external/tensorflow/tensorflow/contrib/estimator/python/estimator/ |
linear.py | 107 def _model_fn(features, labels, mode, config): function in function:LinearEstimator.__init__ 118 model_fn=_model_fn, model_dir=model_dir, config=config)
|
dnn.py | 120 def _model_fn(features, labels, mode, config): function in function:DNNEstimator.__init__ 134 model_fn=_model_fn, model_dir=model_dir, config=config)
|
dnn_linear_combined.py | 147 def _model_fn(features, labels, mode, config): function in function:DNNLinearCombinedEstimator.__init__ 164 model_fn=_model_fn, model_dir=model_dir, config=config)
|
/external/tensorflow/tensorflow/python/estimator/ |
estimator_test.py | 429 def _model_fn(features, labels, mode, params, config): function in function:EstimatorTrainTest.test_callable_input_fn 444 est = estimator.Estimator(model_fn=_model_fn, 457 def _model_fn(features, labels, mode, params, config): function in function:EstimatorTrainTest.test_input_fn_args 468 est = estimator.Estimator(model_fn=_model_fn, 482 def _model_fn(features): function in function:EstimatorTrainTest.test_minimal_model_fn_args 493 est = estimator.Estimator(model_fn=_model_fn) 503 def _model_fn(features): function in function:EstimatorTrainTest.test_labels_should_be_none_if_model_fn_does_not_use_labels 511 est = estimator.Estimator(model_fn=_model_fn) 520 def _model_fn(features): function in function:EstimatorTrainTest.test_input_fn_len_should_be_2_if_tuple_or_list 523 est = estimator.Estimator(model_fn=_model_fn) 539 def _model_fn(mode, params, features, labels, config): function in function:EstimatorTrainTest.test_all_model_fn_args 568 def _model_fn(features, labels, foo, mode, params, config, bar): function in function:EstimatorTrainTest.test_partial_model_fn_args 851 def _model_fn(features, labels, mode): function in function:EstimatorTrainTest.test_features_labels_mode 867 def _model_fn(features, labels, mode): function in function:EstimatorTrainTest.test_graph_initialization_global_step_and_random_seed 917 def _model_fn(features, labels, mode): function in function:EstimatorGetVariablesTest.test_model_should_be_trained 933 def _model_fn(features, labels, mode): function in function:EstimatorGetVariablesTest.test_get_variable_utils 958 def _model_fn(features, labels, mode): function in function:EstimatorDatasetIntegrationTest.test_returned_by_input_fn 974 def _model_fn(features, labels, mode): function in function:EstimatorDatasetIntegrationTest.test_with_none_labels 991 def _model_fn(features, labels, mode): function in function:EstimatorDatasetIntegrationTest.test_with_predict 1009 def _model_fn(features, labels, mode): function in function:EstimatorDatasetIntegrationTest.test_batching 1032 def _model_fn(features, labels, mode, params, config): function in function:EstimatorEvaluateTest.test_input_fn_args 1052 def _model_fn(features, labels, mode): function in function:EstimatorEvaluateTest.test_model_fn_must_return_estimator_spec 1085 def _model_fn(features, labels, mode): function in function:EstimatorEvaluateTest.test_tuple_metrics 1216 def _model_fn(features, labels, mode): function in function:EstimatorEvaluateTest.test_features_labels_mode 1233 def _model_fn(features, labels, mode): function in function:EstimatorEvaluateTest.test_graph_initialization_global_step_and_random_seed 1308 def _model_fn(features, labels, mode, params, config): function in function:EstimatorPredictTest.test_input_fn_args 1347 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_tensor_predictions 1380 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_warn_if_no_queue_runner 1398 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_skip_warn_if_dataset_returns_features 1419 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_skip_warn_if_dataset_returns_features_dict 1441 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_input_fn_can_return_just_features 1459 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_batch_size_mismatch 1478 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_iterate_batches 1499 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_predict_keys_defined_for_tensor 1516 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_predict_keys_does_not_exists 1535 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_return_given_predict_keys 1554 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_yield_rows_of_tensor 1570 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_yield_rows_of_dict 1595 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_hooks_are_used 1617 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_predict_from_old_model_dir 1634 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_predict_from_checkpoint_path 1680 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_features_labels_mode 1697 def _model_fn(features, labels, mode): function in function:EstimatorPredictTest.test_graph_initialization_global_step_and_random_seed 2023 def _model_fn(features, labels, mode): function in function:EstimatorExportTest.test_features_labels_mode 2044 def _model_fn(features, labels, mode): function in function:EstimatorExportTest.test_graph_initialization_global_step_and_random_seed 2237 def _model_fn(features, labels, mode): function in function:EstimatorIntegrationTest.test_complete_flow_with_a_simple_linear_model [all...] |
estimator.py | 215 self._model_fn = model_fn [all...] |
/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/python/estimator/canned/ |
baseline.py | 265 def _model_fn(features, labels, mode, config): function in function:BaselineClassifier.__init__ 275 model_fn=_model_fn, 351 def _model_fn(features, labels, mode, config): function in function:BaselineRegressor.__init__ 360 model_fn=_model_fn,
|
linear.py | 308 def _model_fn(features, labels, mode, config): function in function:LinearClassifier.__init__ 321 model_fn=_model_fn, 429 def _model_fn(features, labels, mode, config): function in function:LinearRegressor.__init__ 442 model_fn=_model_fn,
|
dnn.py | 341 def _model_fn(features, labels, mode, config): function in function:DNNClassifier.__init__ 357 model_fn=_model_fn, model_dir=model_dir, config=config, 485 def _model_fn(features, labels, mode, config): function in function:DNNRegressor.__init__ 504 model_fn=_model_fn, model_dir=model_dir, config=config,
|
dnn_linear_combined.py | 388 def _model_fn(features, labels, mode, config): function in function:DNNLinearCombinedClassifier.__init__ 406 model_fn=_model_fn, model_dir=model_dir, config=config, 551 def _model_fn(features, labels, mode, config): function in function:DNNLinearCombinedRegressor.__init__ 572 model_fn=_model_fn, model_dir=model_dir, config=config,
|
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
dnn_tree_combined_estimator.py | 333 def _model_fn(features, labels, mode, config): function in function:DNNBoostedTreeCombinedClassifier.__init__ 343 model_fn=_model_fn, model_dir=model_dir, 427 def _model_fn(features, labels, mode, config): function in function:DNNBoostedTreeCombinedRegressor.__init__ 436 model_fn=_model_fn, model_dir=model_dir, 504 def _model_fn(features, labels, mode, config): function in function:DNNBoostedTreeCombinedEstimator.__init__ 514 model_fn=_model_fn, model_dir=model_dir,
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
logistic_regressor.py | 45 def _model_fn(features, labels, mode, params): function in function:_get_model_fn_with_logistic_metrics 69 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/tensor_forest/client/ |
random_forest.py | 143 def _model_fn(features, labels, mode): function in function:get_model_fn 251 return _model_fn 417 def _model_fn(features, labels, mode): function in function:get_combined_model_fn 458 return _model_fn
|
/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/gan/python/estimator/python/ |
gan_estimator_impl.py | 156 def _model_fn(features, labels, mode): function in function:GANEstimator.__init__ 169 model_fn=_model_fn, model_dir=model_dir, config=config)
|
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
tpu_estimator.py | 1797 def _model_fn(features, labels, mode, config, params): function in function:TPUEstimator._augment_model_fn [all...] |