/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
checkpoint_input_pipeline_hook_test.py | 74 def _input_fn(): function in function:CheckpointInputPipelineHookTest.testReturnDatasetFromInputFn 79 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)]) 81 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)]) 86 def _input_fn(): function in function:CheckpointInputPipelineHookTest.testBuildIteratorInInputFn 93 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)]) 95 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)]) 100 def _input_fn(): function in function:CheckpointInputPipelineHookTest.testDoNotRestore 105 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)]) 107 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)]) 110 est.train(_input_fn, steps=2 115 def _input_fn(): function in function:CheckpointInputPipelineHookTest.testRaiseErrorIfNoIterator [all...] |
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
dnn_test.py | 312 def _input_fn(): function in function:DNNClassifierTest.testLogisticRegression_MatrixData_Labels1D 327 classifier.fit(input_fn=_input_fn, steps=5) 328 scores = classifier.evaluate(input_fn=_input_fn, steps=1) 367 def _input_fn(num_epochs=None): function in function:DNNClassifierTest.testEstimatorWithCoreFeatureColumns 395 classifier.fit(input_fn=_input_fn, steps=50) 397 scores = classifier.evaluate(input_fn=_input_fn, steps=1) 400 predict_input_fn = functools.partial(_input_fn, num_epochs=1) 411 def _input_fn(num_epochs=None): function in function:DNNClassifierTest.testLogisticRegression_TensorData 440 classifier.fit(input_fn=_input_fn, steps=50) 442 scores = classifier.evaluate(input_fn=_input_fn, steps=1 522 def _input_fn(): function in function:DNNClassifierTest.testMultiClass_MatrixData_Labels1D 563 def _input_fn(num_epochs=None): function in function:DNNClassifierTest.testMultiClassLabelKeys 705 def _input_fn(num_epochs=None): function in function:DNNClassifierTest.testPredict_AsIterableFalse 751 def _input_fn(num_epochs=None): function in function:DNNClassifierTest.testPredict_AsIterable 803 def _input_fn(num_epochs=None): function in function:DNNClassifierTest.testCustomMetrics 872 def _input_fn(num_epochs=None): function in function:DNNClassifierTest.testTrainSaveLoad 919 def _input_fn(num_epochs=None): function in function:DNNClassifierTest.testTrainWithPartitionedVariables 1065 def _input_fn(): function in function:DNNRegressorTest.testRegression_MatrixData_Labels1D 1102 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testRegression_TensorData 1233 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testPredict_AsIterableFalse 1275 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testPredict_AsIterable 1319 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testCustomMetrics 1380 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testCustomMetricsWithMetricSpec 1438 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testTrainSaveLoad 1484 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testTrainWithPartitionedVariables 1533 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testEnableCenteredBias 1572 def _input_fn(num_epochs=None): function in function:DNNRegressorTest.testDisableCenteredBias [all...] |
dnn_linear_combined_test.py | 259 def _input_fn(): function in function:DNNLinearCombinedClassifierTest.testNoDnnHiddenUnits 277 classifier.fit(input_fn=_input_fn, steps=2) 378 def _input_fn(): function in function:DNNLinearCombinedClassifierTest.testLogisticRegression_TensorData 420 classifier.fit(input_fn=_input_fn, steps=100) 421 scores = classifier.evaluate(input_fn=_input_fn, steps=100) 427 def _input_fn(): function in function:DNNLinearCombinedClassifierTest.testEstimatorWithCoreFeatureColumns 466 classifier.fit(input_fn=_input_fn, steps=100) 467 scores = classifier.evaluate(input_fn=_input_fn, steps=100) 473 def _input_fn(): function in function:DNNLinearCombinedClassifierTest.testTrainWithPartitionedVariables 514 classifier.fit(input_fn=_input_fn, steps=100 548 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedClassifierTest.testMultiClassLabelKeys 795 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedClassifierTest.testCustomMetrics 1227 def _input_fn(): function in function:DNNLinearCombinedRegressorTest.testRegression_TensorData 1341 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testPredict_AsIterableFalse 1380 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testPredict_AsIterable 1419 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testCustomMetrics 1481 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testCustomMetricsWithMetricSpec 1541 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testExport 1588 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testTrainSaveLoad 1621 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testTrainWithPartitionedVariables 1673 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testDisableCenteredBias 1711 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testLinearOnly 1742 def _input_fn(num_epochs=None): function in function:DNNLinearCombinedRegressorTest.testDNNOnly [all...] |
debug_test.py | 233 def _input_fn(): function in function:DebugClassifierTest.testLogisticRegression_MatrixData_Labels1D 242 classifier.fit(input_fn=_input_fn, steps=5) 243 scores = classifier.evaluate(input_fn=_input_fn, steps=1) 273 def _input_fn(num_epochs=None): function in function:DebugClassifierTest.testLogisticRegression_TensorData 290 classifier.fit(input_fn=_input_fn, steps=50) 292 scores = classifier.evaluate(input_fn=_input_fn, steps=1) 295 predict_input_fn = functools.partial(_input_fn, num_epochs=1) 342 def _input_fn(): function in function:DebugClassifierTest.testMultiClass_MatrixData_Labels1D 351 classifier.fit(input_fn=_input_fn, steps=200) 352 scores = classifier.evaluate(input_fn=_input_fn, steps=1 463 def _input_fn(num_epochs=None): function in function:DebugClassifierTest.testCustomMetrics 529 def _input_fn(num_epochs=None): function in function:DebugClassifierTest.testTrainSaveLoad 638 def _input_fn(): function in function:DebugRegressorTest.testRegression_MatrixData_Labels1D 666 def _input_fn(num_epochs=None): function in function:DebugRegressorTest.testRegression_TensorData 770 def _input_fn(num_epochs=None): function in function:DebugRegressorTest.testCustomMetrics 824 def _input_fn(num_epochs=None): function in function:DebugRegressorTest.testTrainSaveLoad [all...] |
linear_test.py | 146 def _input_fn(): function in function:LinearClassifierTest.testMultiClass_MatrixData_Labels1D 160 classifier.fit(input_fn=_input_fn, steps=100) 161 scores = classifier.evaluate(input_fn=_input_fn, steps=1) 182 def _input_fn(num_epochs=None): function in function:LinearClassifierTest.testMultiClassLabelKeys 204 classifier.fit(input_fn=_input_fn, steps=50) 206 scores = classifier.evaluate(input_fn=_input_fn, steps=1) 209 predict_input_fn = functools.partial(_input_fn, num_epochs=1) 223 def _input_fn(): function in function:LinearClassifierTest.testLogisticRegression_MatrixData 236 classifier.fit(input_fn=_input_fn, steps=100) 237 scores = classifier.evaluate(input_fn=_input_fn, steps=1 242 def _input_fn(num_epochs=None): function in function:LinearClassifierTest.testEstimatorWithCoreFeatureColumns 269 def _input_fn(): function in function:LinearClassifierTest.testLogisticRegression_MatrixData_Labels1D 363 def _input_fn(num_epochs=None): function in function:LinearClassifierTest.testCustomMetrics 470 def _input_fn(): function in function:LinearClassifierTest.testTrainWithPartitionedVariables 602 def _input_fn(): function in function:LinearClassifierTest.testWeightColumnLoss 1117 def _input_fn(num_epochs=None): function in function:LinearRegressorTest.testRegression_TensorData 1240 def _input_fn(num_epochs=None): function in function:LinearRegressorTest.testPredict_AsIterableFalse 1278 def _input_fn(num_epochs=None): function in function:LinearRegressorTest.testPredict_AsIterable 1319 def _input_fn(num_epochs=None): function in function:LinearRegressorTest.testCustomMetrics 1396 def _input_fn(num_epochs=None): function in function:LinearRegressorTest.testTrainSaveLoad 1436 def _input_fn(num_epochs=None): function in function:LinearRegressorTest.testTrainWithPartitionedVariables 1483 def _input_fn(num_epochs=None): function in function:LinearRegressorTest.testDisableCenteredBias [all...] |
estimator_test.py | 193 def _input_fn(): function in function:_build_estimator_for_export_tests 205 est.fit(input_fn=_input_fn, steps=20) 232 def _input_fn(): function in function:_build_estimator_for_resource_export_test 268 est.fit(input_fn=_input_fn, steps=1) 321 def _input_fn(): function in function:_make_input_fn 325 return _input_fn [all...] |
kmeans_test.py | 417 def _input_fn(): function in function:KMeansCosineDistanceTest.test_predict_kmeans_plus_plus 421 assignments = list(kmeans.predict_cluster_idx(input_fn=_input_fn))
|
/external/tensorflow/tensorflow/contrib/boosted_trees/examples/ |
binary_mnist.py | 58 def _input_fn(): function in function:get_input_fn 71 return _input_fn
|
mnist.py | 53 def _input_fn(): function in function:get_input_fn 66 return _input_fn
|
/external/tensorflow/tensorflow/contrib/predictor/ |
testing_common.py | 83 def _input_fn(): function in function:get_arithmetic_input_fn 102 return _input_fn
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/ |
monitors.py | [all...] |
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
estimator_test.py | 632 def _input_fn(): function in function:CoreGradientBoostedDecisionTreeEstimators.testWeightedCategoricalColumn._make_input_fn 651 return _input_fn
|
/external/tensorflow/tensorflow/contrib/distribute/python/ |
strategy_test_lib.py | 282 # `_input_fn`. This counter is incremented by 1 each time an input_fn is 287 def _input_fn(input_context): function in function:DistributionTestBase._input_fn_to_test_input_context 303 return _input_fn
|
/external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
kmeans_test.py | 450 def _input_fn(): function in function:KMeansCosineDistanceTest.test_predict_kmeans_plus_plus 454 assignments = list(kmeans.predict_cluster_index(input_fn=_input_fn))
|
/external/tensorflow/tensorflow/python/tpu/ |
tpu_estimator.py | 2682 def _input_fn(ctx): function in function:TPUEstimator._call_input_fn [all...] |