Skip to content

Commit 761dfc5

Browse files
Alexander Lyulkovasmorkalov
authored andcommitted
Removed float indices from dnn tests
1 parent 59823be commit 761dfc5

15 files changed

+15
-11
lines changed
0 Bytes
Binary file not shown.
0 Bytes
Binary file not shown.
120 Bytes
Binary file not shown.
96 Bytes
Binary file not shown.

testdata/dnn/onnx/generate_onnx_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2665,7 +2665,7 @@ def __init__(self, *args, **kwargs):
26652665
super(ArgMax, self).__init__()
26662666

26672667
def forward(self, x):
2668-
return torch.argmax(x, dim=2, keepdims=False).to(torch.float32)
2668+
return torch.argmax(x, dim=2, keepdims=False)
26692669

26702670
model = ArgMax()
26712671
input_ = Variable(torch.randn(2, 3, 4, 5, dtype=torch.float32))
@@ -2676,7 +2676,7 @@ def __init__(self, *args, **kwargs):
26762676
super(ArgMin, self).__init__()
26772677

26782678
def forward(self, x):
2679-
return torch.argmin(x, dim=-1, keepdims=True).to(torch.float32)
2679+
return torch.argmin(x, dim=-1, keepdims=True)
26802680

26812681
model = ArgMin()
26822682
input_ = Variable(torch.randn(2, 3, 4, 5, dtype=torch.float32))

testdata/dnn/onnx/models/argmax.onnx

13 Bytes
Binary file not shown.

testdata/dnn/onnx/models/argmin.onnx

13 Bytes
Binary file not shown.

testdata/dnn/tensorflow/argmax_in.npy

0 Bytes
Binary file not shown.

testdata/dnn/tensorflow/argmax_net.pb

831 Bytes
Binary file not shown.
24 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)