| /external/tensorflow/tensorflow/python/training/ |
| D | basic_loops_test.py | 45 def train_fn(unused_sess, sv, y, a): function
55 sv, train_fn, args=(sv, "y"), kwargs={"a": "A"})
61 def train_fn(unused_sess): function
62 train_fn.counter += 1
63 if train_fn.counter == 3:
67 train_fn.counter = 0
72 basic_loops.basic_train_loop(sv, train_fn)
83 def train_fn(self, unused_sess): member in BasicTrainLoopTest.testBasicTrainLoopRetryOnAborted.AbortAndRetry
96 basic_loops.basic_train_loop(sv, aar.train_fn)
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | tracking_util_with_v1_optimizers_test.py | 402 train_fn = functools.partial(
407 train_fn = functools.partial(self.evaluate, train_fn())
410 train_fn()
435 def train_fn(): function
445 train_fn = functools.partial(
446 self.evaluate, train_fn())
449 train_fn()
510 train_fn = functools.partial(
515 train_fn = functools.partial(self.evaluate, train_fn())
518 train_fn()
[all …]
|
| D | tracking_util_test.py | 356 train_fn = functools.partial(_train_fn, model, input_value)
358 train_fn = functools.partial(self.evaluate, train_fn())
361 train_fn()
410 def train_fn(): function
419 train_fn = functools.partial(
420 self.evaluate, train_fn())
423 train_fn()
621 def train_fn(): function
628 train_fn = functools.partial(self.evaluate, train_fn())
638 train_fn()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_optimizer_v2_test.py | 72 def train_fn(): function
83 train_fn = sess.make_callable(train_fn())
87 train_fn()
97 train_fn()
|
| D | custom_training_loop_metrics_test.py | 113 def train_fn(dataset): function
122 train_fn(dataset)
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | fault_tolerance_test.py | 86 def train_fn(self, iterator): member in Model
95 self.cluster_coord.schedule(self.train_fn, args=(self.iterator,))
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | custom_training_loop_input_test.py | 172 def train_fn(distributed_iterator): function
183 train_fn(distributed_iterator)
|
| D | tpu_strategy_test.py | 295 def train_fn(iterator): function
305 logging.info(train_fn(iterator))
|
| D | input_lib_test.py | 898 def train_fn(): function
910 train_fn()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks.py | 2220 train_fn = self.model.train_function
2222 if hasattr(train_fn, 'function_spec'):
2223 … summary_ops_v2.graph(train_fn._concrete_stateful_fn.graph) # pylint: disable=protected-access
|