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