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