diff --git a/circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst b/circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst index afeb1d01aec..8186ef038b8 100644 --- a/circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst +++ b/circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst @@ -48,6 +48,8 @@ AddModel(ConvTranspose2d_F32_R4_op01) AddModel(ConvTranspose2d_F32_R4_p10) AddModel(ConvTranspose2d_F32_R4_p11) AddModel(ConvTranspose2d_F32_R4_p11_nobias) +# AddModel(ConvTranspose2d_F32_R4_unk_bh) +# AddModel(ConvTranspose2d_F32_R4_unk_bw) AddModel(Cos_F32_R4) AddModel(CumSum_F32_R4_1) AddModel(CumSum_F32_R4_2) diff --git a/circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst b/circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst index 05c3f9131ec..5e8c623b9f0 100644 --- a/circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst +++ b/circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst @@ -50,6 +50,8 @@ AddModel(ConvTranspose2d_F32_R4_op01) AddModel(ConvTranspose2d_F32_R4_p10) AddModel(ConvTranspose2d_F32_R4_p11) AddModel(ConvTranspose2d_F32_R4_p11_nobias) +# AddModel(ConvTranspose2d_F32_R4_unk_bh) +# AddModel(ConvTranspose2d_F32_R4_unk_bw) AddModel(Cos_F32_R4) AddModel(CumSum_F32_R4_1) AddModel(CumSum_F32_R4_2) diff --git a/circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst b/circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst index d7feef42a74..9133268ffaa 100644 --- a/circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst +++ b/circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst @@ -50,6 +50,8 @@ AddModel(ConvTranspose2d_F32_R4_op01) AddModel(ConvTranspose2d_F32_R4_p10) AddModel(ConvTranspose2d_F32_R4_p11) AddModel(ConvTranspose2d_F32_R4_p11_nobias) +# AddModel(ConvTranspose2d_F32_R4_unk_bh) +# AddModel(ConvTranspose2d_F32_R4_unk_bw) AddModel(Cos_F32_R4) AddModel(CumSum_F32_R4_1) AddModel(CumSum_F32_R4_2) diff --git a/circle-mlir/models/unit/ConvTranspose2d_F32_R4_unk_bh/__init__.py b/circle-mlir/models/unit/ConvTranspose2d_F32_R4_unk_bh/__init__.py new file mode 100644 index 00000000000..3a2f97581fb --- /dev/null +++ b/circle-mlir/models/unit/ConvTranspose2d_F32_R4_unk_bh/__init__.py @@ -0,0 +1,35 @@ +import torch + + +# Generate ConvTranspose2d operator with Float32, Rank-4, unknown +# input : [N, 4, H, 10] +# output : [N, 3, H, 10+7] +# dynamic axes: N, H +class net_ConvTranspose2d(torch.nn.Module): + def __init__(self): + super().__init__() + self.op = torch.nn.ConvTranspose2d( + in_channels=4, + out_channels=3, + kernel_size=(1, 8), + stride=(1, 1), + padding=(0, 0), + dilation=(1, 1), + groups=1, + bias=True, + ) + + def forward(self, input): + return self.op(input) + + def onnx_opset_version(self): + # TODO set to appropriate value + return 14 + + +_model_ = net_ConvTranspose2d() + +_inputs_ = (torch.Tensor(1, 4, 1, 1)) + +_io_names_ = [['input'], ['output']] +_dynamic_axes_ = {"input": {0: "?", 2: "?"}, "output": {0: "?", 2: "?"}} diff --git a/circle-mlir/models/unit/ConvTranspose2d_F32_R4_unk_bw/__init__.py b/circle-mlir/models/unit/ConvTranspose2d_F32_R4_unk_bw/__init__.py new file mode 100644 index 00000000000..ff4d82e020a --- /dev/null +++ b/circle-mlir/models/unit/ConvTranspose2d_F32_R4_unk_bw/__init__.py @@ -0,0 +1,35 @@ +import torch + + +# Generate ConvTranspose2d operator with Float32, Rank-4, unknown +# input : [N, 4, 1, W] +# output : [N, 3, 1, W+7] +# dynamic axes: N, W +class net_ConvTranspose2d(torch.nn.Module): + def __init__(self): + super().__init__() + self.op = torch.nn.ConvTranspose2d( + in_channels=4, + out_channels=3, + kernel_size=(1, 8), + stride=(1, 1), + padding=(0, 0), + dilation=(1, 1), + groups=1, + bias=True, + ) + + def forward(self, input): + return self.op(input) + + def onnx_opset_version(self): + # TODO set to appropriate value + return 14 + + +_model_ = net_ConvTranspose2d() + +_inputs_ = (torch.Tensor(1, 4, 1, 1)) + +_io_names_ = [['input'], ['output']] +_dynamic_axes_ = {"input": {0: "?", 3: "?"}, "output": {0: "?", 3: "?"}}