Skip to content

Commit 77c217c

Browse files
committed
Add ReduceMax testdata
1 parent 207eacd commit 77c217c

File tree

4 files changed

+9
-0
lines changed

4 files changed

+9
-0
lines changed
176 Bytes
Binary file not shown.
132 Bytes
Binary file not shown.

testdata/dnn/onnx/generate_onnx_models.py

+9
Original file line numberDiff line numberDiff line change
@@ -943,3 +943,12 @@ def forward(self, x):
943943
x = Variable(torch.randn(2, 2))
944944
model = Power(2)
945945
save_data_and_model("pow2", x, model)
946+
947+
class ReduceMax(nn.Module):
948+
def forward(self, x):
949+
out = torch.max(x)
950+
return torch.unsqueeze(out, 0)
951+
952+
x = Variable(torch.randn(1, 3, 2, 2))
953+
model = ReduceMax()
954+
save_data_and_model("reduce_max", x, model)
181 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)