def METHOD_NAME(program_config, predictor_config):
    if "nvidia_tensorrt" in self.get_nnadapter_device_name():
        in_shape = program_config.inputs["stack_input1"].shape
        axis = program_config.ops[0].attrs["axis"]
        if len(in_shape) == 1 or axis == 0:
            return True