Searched defs:on_train_batch_end (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A 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
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
|
| H A D | callbacks_v1.py | 350 def on_train_batch_end(self, batch, logs=None): member in TensorBoard
|