Searched refs:train_function (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 612 self.train_function = None
835 if self.train_function is not None:
836 return self.train_function
857 def train_function(iterator): function
863 def train_function(iterator): function
870 train_function = def_function.function(
871 train_function, experimental_relax_shapes=True)
872 self.train_tf_function = train_function
874 self.train_function = train_function
877 …self.train_function = lambda iterator: self._cluster_coordinator.schedule( # pylint: disable=g-lo…
[all …]
|
| D | training_v1.py | 465 self.train_function = None
1090 outputs = self.train_function(ins) # pylint: disable=not-callable
2094 return self.train_function
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | mnist_conv_custom_training_benchmark_test.py | 117 def train_function(self, member in CustomMnistBenchmark
235 self.train_function(model, dataset, loss_fn, optimizer, 1,
240 train_step_time = self.train_function(model, dataset, loss_fn, optimizer,
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saved_model_experimental.py | 286 train_op = clone.train_function.updates_op
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 679 model.train_function = None
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks_test.py | 1931 self.model.train_function = new_fn
|