KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16]
request = <FixtureRequest for <Function test_acosh[in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    def test_acosh(in_shape, request):
        model_def = _make_module(in_shape)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_acosh.py:93: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7f12f4c48a10>
case_name = 'test_acosh_in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [[1, 3, 16, 16], [1]]
request = <FixtureRequest for <Function test_and[in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    def test_and(in_shape, request):
        model_def = _make_module(in_shape[0], in_shape[1])
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_and.py:79: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456deb10>
case_name = 'test_and_in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [[1, 3, 16, 16], [16]]
request = <FixtureRequest for <Function test_and[in_shape1]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    def test_and(in_shape, request):
        model_def = _make_module(in_shape[0], in_shape[1])
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_and.py:79: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456cf650>
case_name = 'test_and_in_shape1_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [[1, 3, 16, 16], [1, 16]]
request = <FixtureRequest for <Function test_and[in_shape2]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    def test_and(in_shape, request):
        model_def = _make_module(in_shape[0], in_shape[1])
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_and.py:79: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456fa350>
case_name = 'test_and_in_shape2_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [[1, 3, 16, 16], [1, 16, 16]]
request = <FixtureRequest for <Function test_and[in_shape3]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    def test_and(in_shape, request):
        model_def = _make_module(in_shape[0], in_shape[1])
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_and.py:79: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed14577d650>
case_name = 'test_and_in_shape3_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [[1, 1, 16, 16], [3, 3, 1, 16]]
request = <FixtureRequest for <Function test_and[in_shape4]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    def test_and(in_shape, request):
        model_def = _make_module(in_shape[0], in_shape[1])
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_and.py:79: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456ec190>
case_name = 'test_and_in_shape4_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = None, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None-None-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7f12100167d0>
case_name = 'test_argmax_None-None-None-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = 0, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None-0-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed154090150>
case_name = 'test_argmax_None-None-0-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = 1, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None-1-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456e7110>
case_name = 'test_argmax_None-None-1-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = 2, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None-2-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456e0050>
case_name = 'test_argmax_None-None-2-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = 3, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None-3-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456de2d0>
case_name = 'test_argmax_None-None-3-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = -1, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None--1-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456fa9d0>
case_name = 'test_argmax_None-None--1-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = -2, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None--2-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed14572e790>
case_name = 'test_argmax_None-None--2-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = -3, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None--3-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed145715590>
case_name = 'test_argmax_None-None--3-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = -4, keepdim = None, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-None--4-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456f2d10>
case_name = 'test_argmax_None-None--4-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = None, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True-None-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed145706850>
case_name = 'test_argmax_None-True-None-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = 0, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True-0-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456cc6d0>
case_name = 'test_argmax_None-True-0-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = 1, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True-1-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456e0910>
case_name = 'test_argmax_None-True-1-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = 2, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True-2-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456ec8d0>
case_name = 'test_argmax_None-True-2-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = 3, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True-3-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed145715fd0>
case_name = 'test_argmax_None-True-3-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = -1, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True--1-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed14575f650>
case_name = 'test_argmax_None-True--1-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = -2, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True--2-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed1456cfdd0>
case_name = 'test_argmax_None-True--2-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = -3, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True--3-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed14572a390>
case_name = 'test_argmax_None-True--3-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = -4, keepdim = True, select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-True--4-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed14573d490>
case_name = 'test_argmax_None-True--4-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError

KeyError: 'dump_infer'
in_shape = [1, 3, 16, 16], axis = None, keepdim = False
select_last_index = None
request = <FixtureRequest for <Function test_argmax[None-False-None-in_shape0]>>

    @pytest.mark.parametrize('in_shape', in_shapes)
    @pytest.mark.parametrize('axis', axes)
    @pytest.mark.parametrize('keepdim', keepdims)
    @pytest.mark.parametrize('select_last_index', select_last_indices)
    def test_argmax(in_shape, axis, keepdim, select_last_index, request):
        model_def = _make_module(in_shape, axis, keepdim, select_last_index)
    
>       runner = OnnxTestRunner(request.node.name)

tests/importer/onnx_/basic/test_argmax.py:111: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
tests/onnx_test_runner.py:16: in __init__
    super().__init__(case_name, overwrite_configs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <onnx_test_runner.OnnxTestRunner object at 0x7ed145704d50>
case_name = 'test_argmax_None-False-None-in_shape0_', override_cfg = None

    def __init__(self, case_name, override_cfg: str = None) -> None:
        config_root = os.path.dirname(__file__)
        default_cfg = toml.load(os.path.join(config_root, 'config.toml'))
    
        new_cfg = None
        if override_cfg is not None:
            if os.path.isfile(override_cfg):
                new_cfg = toml.load(override_cfg)
            else:
                new_cfg = toml.loads(override_cfg)
        config = self.update_config(default_cfg, new_cfg)
        self.cfg = self.validte_config(config)
    
        case_name = case_name.replace('[', '_').replace(']', '_')
        self.case_dir = os.path.join(self.cfg['root'], case_name)
        self.clear(self.case_dir)
        os.makedirs(self.case_dir)
    
        self.inputs: List[Dict] = []
        self.calibs: List[Dict] = []
        self.outputs: List[Dict] = []
        self.model_type: str = ""
        self.pre_process: List[Dict] = []
    
        self.num_pattern = re.compile("(\d+)")
        # [n, c, h, w].zip default_shape => [(n, 1), (c, 1), (h, 48), (w, 48)]
        self.default_shape = [1, 1, 48, 48, 24, 24]
        self.shape_vars = {}
        # used for tag dynamic model for onnx simplify
        self.dynamic = False
    
>       if self.cfg['dump_infer']:
E       KeyError: 'dump_infer'

tests/test_runner.py:58: KeyError