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