HomeSort by relevance Sort by last modified time
    Searched refs:_input_fn (Results 1 - 15 of 15) sorted by null

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

Completed in 2062 milliseconds