Searched defs:dist_dataset (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | custom_training_loop_input_test.py | 312 def f_train_step(dist_dataset): argument |
| D | input_lib_test.py | 1006 def iterator_fn(dist_dataset): argument |