benchmarks/layer_benchmark/base_benchmark.py/KerasCoreBenchmarkMetricsCallback/on_predict_batch_end
class KerasCoreBenchmarkMetricsCallback:    def on_predict_batch_end(self, batch, logs=None):
        self._callback.on_predict_batch_end(batch, logs)