Searched defs:train_fn (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ | ||
| D | basic_loops_test.py | 41 def train_fn(unused_sess, sv, y, a): function |
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | tpu_strategy_model_parallelism_test.py | 212 def train_fn(batch_features): function |
| D | input_lib_test.py | 876 def train_fn(): function |
| D | custom_training_loop_input_test.py | 170 def train_fn(distributed_iterator): function |
| D | tpu_strategy_test.py | 432 def train_fn(iterator): function |
| /external/tensorflow/tensorflow/python/distribute/coordinator/ | ||
| D | fault_tolerance_test.py | 103 def train_fn(self, iterator, iterator2): member in Model |