Searched refs:on_train_batch_end (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | dataset_creator_model_fit_ps_only_test.py | 67 def on_train_batch_end(self, batch, logs=None): member in DatasetCreatorModelFitParameterServerStrategyOnlyTest.testModelFitErrorOnBatchLevelCallbacks.BatchLevelCallback
84 def on_train_batch_end(self, batch, logs=None): member in DatasetCreatorModelFitParameterServerStrategyOnlyTest.testModelFitCallbackSupportsTFLogs.MyCallback
123 def on_train_batch_end(self, batch, logs=None): member in DatasetCreatorModelFitParameterServerStrategyOnlyTest.testModelEvaluateErrorOnBatchLevelCallbacks.BatchLevelCallback
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks.py | 448 def on_train_batch_end(self, batch, logs=None): member in CallbackList
728 def on_train_batch_end(self, batch, logs=None): member in Callback
891 not generic_utils.is_default(self.on_train_batch_end))
1051 def on_train_batch_end(self, batch, logs=None): member in ProgbarLogger
1381 def on_train_batch_end(self, batch, logs=None): member in ModelCheckpoint
2439 def on_train_batch_end(self, batch, logs=None): member in TensorBoard
|
| D | callbacks_v1.py | 350 def on_train_batch_end(self, batch, logs=None): member in TensorBoard
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 1192 callbacks.on_train_batch_end(end_step, logs)
|