Searched refs:get_input_fn (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/examples/ |
| D | mnist.py | 47 def get_input_fn(dataset_split, function
104 train_input_fn = get_input_fn(data.train, FLAGS.batch_size)
105 eval_input_fn = get_input_fn(data.validation, FLAGS.eval_batch_size)
|
| D | binary_mnist.py | 46 def get_input_fn(data, function
102 train_input_fn = get_input_fn(data.train, FLAGS.batch_size)
103 eval_input_fn = get_input_fn(data.validation, FLAGS.eval_batch_size)
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | main_estimator.py | 92 def get_input_fn(config, data_dir, split): function
161 train_input_fn = get_input_fn(
163 eval_input_fn = get_input_fn(
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dynamic_rnn_estimator_test.py | 269 def get_input_fn(mode): function
296 estimator_fn().fit(input_fn=get_input_fn(model_fn_lib.ModeKeys.TRAIN),
305 input_fn=get_input_fn(model_fn_lib.ModeKeys.INFER),
424 def get_input_fn(batch_size, sequence_length, state_dict, starting_step=0): function
460 train_input_fn = get_input_fn(
470 input_fn = get_input_fn(
|
| D | state_saving_rnn_estimator_test.py | 392 def get_input_fn(mode, seed): function
428 estimator_fn().fit(input_fn=get_input_fn(
438 input_fn=get_input_fn(
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | gmm_test.py | 190 def get_input_fn(x): function
195 gmm.fit(input_fn=get_input_fn(x), steps=iterations)
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/g3doc/ |
| D | tutorial.md | 55 def get_input_fn(dataset_split, batch_size, capacity=10000, min_after_dequeue=3000):
72 train_input_fn = get_input_fn(data.train, batch_size=256)
73 eval_input_fn = get_input_fn(data.validation, batch_size=5000)
|
| /external/tensorflow/tensorflow/contrib/tpu/ |
| D | tpu_estimator.md | 68 def get_input_fn(filename):
116 estimator.train(input_fn=get_input_fn(FLAGS.train_file),
|