Searched refs:_call_batch_hook (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/keras/engine/ |
D | training_distributed_v1.py | 232 callbacks._call_batch_hook(mode, 'begin', step_index, batch_logs) 247 callbacks._call_batch_hook(mode, 'end', step_index, batch_logs) 379 callbacks._call_batch_hook(mode, 'begin', current_step, batch_logs) 400 callbacks._call_batch_hook(mode, 'end', current_step, batch_logs) 525 callbacks._call_batch_hook(mode, 'begin', current_step, batch_logs) 548 callbacks._call_batch_hook(mode, 'end', current_step, batch_logs)
|
D | training_arrays_v1.py | 279 callbacks._call_batch_hook(mode, 'begin', step, batch_logs) 336 callbacks._call_batch_hook(mode, 'end', step, batch_logs) 377 callbacks._call_batch_hook(mode, 'begin', batch_index, batch_logs) 391 callbacks._call_batch_hook(mode, 'end', batch_index, batch_logs)
|
D | training_generator_v1.py | 253 callbacks._call_batch_hook(mode, 'begin', step, batch_logs) 282 callbacks._call_batch_hook(mode, 'end', step, batch_logs)
|
/external/tensorflow/tensorflow/python/keras/ |
D | callbacks.py | 308 def _call_batch_hook(self, mode, hook, batch, logs=None): member in CallbackList 401 self._call_batch_hook(ModeKeys.TRAIN, 'begin', batch, logs=logs) 405 self._call_batch_hook(ModeKeys.TRAIN, 'end', batch, logs=logs) 446 self._call_batch_hook(ModeKeys.TRAIN, 'begin', batch, logs=logs) 456 self._call_batch_hook(ModeKeys.TRAIN, 'end', batch, logs=logs) 468 self._call_batch_hook(ModeKeys.TEST, 'begin', batch, logs=logs) 478 self._call_batch_hook(ModeKeys.TEST, 'end', batch, logs=logs) 490 self._call_batch_hook(ModeKeys.PREDICT, 'begin', batch, logs=logs) 500 self._call_batch_hook(ModeKeys.PREDICT, 'end', batch, logs=logs)
|