def METHOD_NAME(self, inputs: list, **kwargs):
    """Rewrite this func to remove voxelize op.
    Args:
        inputs (list): input list comprises voxels, num_points and coors
    Returns:
        tuple: A tuple of classification scores, bbox and direction
            classification prediction.
            - cls_scores (list[Tensor]): Classification scores for all
                scale levels, each is a 4D-tensor, the channels number
                is num_base_priors * num_classes.
            - bbox_preds (list[Tensor]): Box energies / deltas for all
                scale levels, each is a 4D-tensor, the channels number
                is num_base_priors * C.
            - dir_cls_preds (list[Tensor|None]): Direction classification
                predictions for all scale levels, each is a 4D-tensor,
                the channels number is num_base_priors * 2.
    """
    ctx = FUNCTION_REWRITER.get_context()
    deploy_cfg = ctx.cfg
    batch_inputs_dict = {
        'voxels': {
            'voxels': inputs[0],
            'num_points': inputs[1],
            'coors': inputs[2]
        }
    }
    _, pts_feats = self.extract_feat(batch_inputs_dict=batch_inputs_dict)
    outs = self.pts_bbox_head(pts_feats)
    if type(outs[0][0]) is dict:
        bbox_preds, scores, dir_scores = [], [], []
        for task_res in outs:
            bbox_preds.append(task_res[0]['reg'])
            bbox_preds.append(task_res[0]['height'])
            bbox_preds.append(task_res[0]['dim'])
            if 'vel' in task_res[0].keys():
                bbox_preds.append(task_res[0]['vel'])
            scores.append(task_res[0]['heatmap'])
            dir_scores.append(task_res[0]['rot'])
        bbox_preds = torch.cat(bbox_preds, dim=1)
        scores = torch.cat(scores, dim=1)
        dir_scores = torch.cat(dir_scores, dim=1)
        return scores, bbox_preds, dir_scores
    else:
        preds = []
        expect_names = []
        for i in range(len(outs[0])):
            preds += [outs[0][i], outs[1][i], outs[2][i]]
            expect_names += [
                f'cls_score{i}', f'bbox_pred{i}', f'dir_cls_pred{i}'
            ]
        # check if output_names is set correctly.
        onnx_cfg = get_ir_config(deploy_cfg)
        output_names = onnx_cfg['output_names']
        if output_names != list(expect_names):
            raise RuntimeError(f'`output_names` should be {expect_names} '
                               f'but given {output_names}\n'
                               f'Deploy config:\n{deploy_cfg.pretty_text}')
        return tuple(preds)