diff --git a/testdata/dnn/onnx/data/input_sub_from_const1.npy b/testdata/dnn/onnx/data/input_sub_from_const1.npy new file mode 100644 index 000000000..7ebf75d8a Binary files /dev/null and b/testdata/dnn/onnx/data/input_sub_from_const1.npy differ diff --git a/testdata/dnn/onnx/data/input_sub_from_const_broadcast.npy b/testdata/dnn/onnx/data/input_sub_from_const_broadcast.npy new file mode 100644 index 000000000..5f34483a0 Binary files /dev/null and b/testdata/dnn/onnx/data/input_sub_from_const_broadcast.npy differ diff --git a/testdata/dnn/onnx/data/input_sub_from_const_eltwise.npy b/testdata/dnn/onnx/data/input_sub_from_const_eltwise.npy new file mode 100644 index 000000000..38408b6ed Binary files /dev/null and b/testdata/dnn/onnx/data/input_sub_from_const_eltwise.npy differ diff --git a/testdata/dnn/onnx/data/output_sub_from_const1.npy b/testdata/dnn/onnx/data/output_sub_from_const1.npy new file mode 100644 index 000000000..a4a4aa667 Binary files /dev/null and b/testdata/dnn/onnx/data/output_sub_from_const1.npy differ diff --git a/testdata/dnn/onnx/data/output_sub_from_const_broadcast.npy b/testdata/dnn/onnx/data/output_sub_from_const_broadcast.npy new file mode 100644 index 000000000..c7ad16fec Binary files /dev/null and b/testdata/dnn/onnx/data/output_sub_from_const_broadcast.npy differ diff --git a/testdata/dnn/onnx/data/output_sub_from_const_eltwise.npy b/testdata/dnn/onnx/data/output_sub_from_const_eltwise.npy new file mode 100644 index 000000000..013f17ad3 Binary files /dev/null and b/testdata/dnn/onnx/data/output_sub_from_const_eltwise.npy differ diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py index af8032550..d1965c2f4 100644 --- a/testdata/dnn/onnx/generate_onnx_models.py +++ b/testdata/dnn/onnx/generate_onnx_models.py @@ -1523,3 +1523,38 @@ def save_data_and_paddle_model(model, name, input_data): input_shape = [1, 2, 3, 4] x = paddle.rand(input_shape, dtype="float32") save_data_and_paddle_model(Resize_HumanSeg(), "resize_humanseg", x) + +class SubFromConstBroadcast(nn.Module): + def __init__(self, *args, **kwargs): + super(SubFromConstBroadcast, self).__init__() + self.const = torch.randn(1, 3, dtype=torch.float32) + + def forward(self, x): + return self.const - x + +model = SubFromConstBroadcast() +input_ = Variable(torch.randn(2, 3, dtype=torch.float32)) +save_data_and_model("sub_from_const_broadcast", input_, model) + +class SubFromConstEltWise(nn.Module): + def __init__(self, *args, **kwargs): + super(SubFromConstEltWise, self).__init__() + self.const = torch.randn(1, 2, 3, 4, dtype=torch.float32) + + def forward(self, x): + return self.const - x + +model = SubFromConstEltWise() +input_ = Variable(torch.randn(1, 2, 3, 4, dtype=torch.float32)) +save_data_and_model("sub_from_const_eltwise", input_, model) + +class SubFromConst1(nn.Module): + def __init__(self, *args, **kwargs): + super(SubFromConst1, self).__init__() + + def forward(self, x): + return 1 - x + +model = SubFromConst1() +input_ = Variable(torch.randn(1, 2, 3, 4, dtype=torch.float32)) +save_data_and_model("sub_from_const1", input_, model) diff --git a/testdata/dnn/onnx/models/sub_from_const1.onnx b/testdata/dnn/onnx/models/sub_from_const1.onnx new file mode 100644 index 000000000..4809de5f2 Binary files /dev/null and b/testdata/dnn/onnx/models/sub_from_const1.onnx differ diff --git a/testdata/dnn/onnx/models/sub_from_const_broadcast.onnx b/testdata/dnn/onnx/models/sub_from_const_broadcast.onnx new file mode 100644 index 000000000..feabfa849 --- /dev/null +++ b/testdata/dnn/onnx/models/sub_from_const_broadcast.onnx @@ -0,0 +1,15 @@ +pytorch1.9: +;1 +Constant_0"Constant* +value*J W@@-oy + +1 +02Sub_1"Subtorch-jit-exportZ +0 +  + +b +2 +  + +B \ No newline at end of file diff --git a/testdata/dnn/onnx/models/sub_from_const_eltwise.onnx b/testdata/dnn/onnx/models/sub_from_const_eltwise.onnx new file mode 100644 index 000000000..feb661f7c --- /dev/null +++ b/testdata/dnn/onnx/models/sub_from_const_eltwise.onnx @@ -0,0 +1,19 @@ +pytorch1.9: +1 +Constant_0"Constant*x +value*lJ`)hVg>]Dt8 /0?Fko]3uI?X=.&,[W>aZ]ǒ?2V־݂ + +1 +02Sub_1"Subtorch-jit-exportZ +0 + + + + +b +2 + + + + +B \ No newline at end of file