Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/
Dbasic_loops_test.py45 def train_fn(unused_sess, sv, y, a): function
61 def train_fn(unused_sess): function
83 def train_fn(self, unused_sess): member in BasicTrainLoopTest.testBasicTrainLoopRetryOnAborted.AbortAndRetry
/external/tensorflow/tensorflow/python/keras/distribute/
Dkeras_optimizer_v2_test.py72 def train_fn(): function
Dcustom_training_loop_metrics_test.py113 def train_fn(dataset): function
/external/tensorflow/tensorflow/python/keras/tests/
Dtracking_util_test.py410 def train_fn(): function
621 def train_fn(): function
Dtracking_util_with_v1_optimizers_test.py435 def train_fn(): function
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dfault_tolerance_test.py86 def train_fn(self, iterator): member in Model
/external/tensorflow/tensorflow/python/distribute/
Dcustom_training_loop_input_test.py172 def train_fn(distributed_iterator): function
Dtpu_strategy_test.py295 def train_fn(iterator): function
Dinput_lib_test.py898 def train_fn(): function