Searched refs:train_step_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | learning.py | 64 train_step_fn=train_step, argument
160 total_loss, should_stop = train_step_fn(sess, train_op, global_step,
168 train_step_fn=train_step_with_pruning_fn,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | basic_loops.py | 25 def basic_train_loop(supervisor, train_step_fn, args=None, argument
57 train_step_fn(sess, *args, **kwargs)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | learning.py | 533 train_step_fn=train_step, argument
769 total_loss, should_stop = train_step_fn(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 269 …argspec: "args=[\'supervisor\', \'train_step_fn\', \'args\', \'kwargs\', \'master\'], varargs=None…
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train_test.py | 1071 train_step_fn=train.get_sequential_train_steps(train_steps))
|