def METHOD_NAME(self, data):
    with paddle.amp.auto_cast(
            level=self.amp_level,
            enable=True,
            custom_white_list={
                'elementwise_add', 'batch_norm', 'sync_batch_norm'
            },
            custom_black_list={'bilinear_interp_v2'}):
        pp_out = self.runner.infer_forward(data)
    return pp_out