Alaa Ali

Add ConstantArgument support to fx_import

1

This PR will fix the following issue: [fx_importer NotImplementedError: MultiheadAttention layer with NeedWeight = false](https://github.com/llvm/torch-mlir/issues/4158) The following error was raised before this fix: Python Error: NotImplementedError: OutputKind.USER_OUTPUT for : ConstantArgument(name='',...

This issue explains a bug in torch-mlir/blob/main/python/torch_mlir/extras/**fx_importer.py** We found this while importing an exported model into MLIR. This occurs for an exported MultiheadAttention layer with "NeedWeight = false" which means...

This code snippet of pixel_unshuffle.mlir that will reproduce the issue: ``` module { func.func @main(%arg0: !torch.vtensor) -> !torch.vtensor attributes {torch.assume_strict_symbolic_shapes} { %int1 = torch.constant.int 1 %0 = torch.operator "torch.aten.pixel_unshuffle"(%arg0, %int1)...