Home
last modified time | relevance | path

Searched defs:train_fn (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dbasic_loops_test.py41 def train_fn(unused_sess, sv, y, a): function
57 def train_fn(unused_sess): function
79 def train_fn(self, unused_sess): member in BasicTrainLoopTest.testBasicTrainLoopRetryOnAborted.AbortAndRetry
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_model_parallelism_test.py212 def train_fn(batch_features): function
Dinput_lib_test.py876 def train_fn(): function
915 def train_fn(): function
955 def train_fn(): function
Dcustom_training_loop_input_test.py170 def train_fn(distributed_iterator): function
Dtpu_strategy_test.py432 def train_fn(iterator): function
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dfault_tolerance_test.py103 def train_fn(self, iterator, iterator2): member in Model