Searched refs:train_function (Results 1 – 4 of 4) sorted by relevance
610 self.train_function = None833 if self.train_function is not None:834 return self.train_function855 def train_function(iterator): function861 def train_function(iterator): function868 train_function = def_function.function(869 train_function, experimental_relax_shapes=True)870 self.train_tf_function = train_function872 self.train_function = train_function875 …self.train_function = lambda iterator: self._cluster_coordinator.schedule( # pylint: disable=g-lo…[all …]
465 self.train_function = None1090 outputs = self.train_function(ins) # pylint: disable=not-callable2094 return self.train_function
285 train_op = clone.train_function.updates_op
679 model.train_function = None