Skip to content

Commit 0839233

Browse files
committed
fix expected data
1 parent cfcabf0 commit 0839233

5 files changed

+25
-0
lines changed

test/onnx/expect/TestOperators.test_softmaxcrossentropy.expect

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,11 @@ graph {
88
output: "2"
99
name: "SoftmaxCrossEntropyLoss_0"
1010
op_type: "SoftmaxCrossEntropyLoss"
11+
attribute {
12+
name: "ignore_index"
13+
i: -100
14+
type: INT
15+
}
1116
attribute {
1217
name: "reduction"
1318
s: "mean"

test/onnx/expect/TestOperators.test_softmaxcrossentropy_3d.expect

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,11 @@ graph {
88
output: "2"
99
name: "SoftmaxCrossEntropyLoss_0"
1010
op_type: "SoftmaxCrossEntropyLoss"
11+
attribute {
12+
name: "ignore_index"
13+
i: -100
14+
type: INT
15+
}
1116
attribute {
1217
name: "reduction"
1318
s: "mean"

test/onnx/expect/TestOperators.test_softmaxcrossentropy_3d_none.expect

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,11 @@ graph {
88
output: "2"
99
name: "SoftmaxCrossEntropyLoss_0"
1010
op_type: "SoftmaxCrossEntropyLoss"
11+
attribute {
12+
name: "ignore_index"
13+
i: -100
14+
type: INT
15+
}
1116
attribute {
1217
name: "reduction"
1318
s: "none"

test/onnx/expect/TestOperators.test_softmaxcrossentropy_4d.expect

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,11 @@ graph {
88
output: "2"
99
name: "SoftmaxCrossEntropyLoss_0"
1010
op_type: "SoftmaxCrossEntropyLoss"
11+
attribute {
12+
name: "ignore_index"
13+
i: -100
14+
type: INT
15+
}
1116
attribute {
1217
name: "reduction"
1318
s: "mean"

test/onnx/expect/TestOperators.test_softmaxcrossentropy_weights.expect

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,11 @@ graph {
99
output: "3"
1010
name: "SoftmaxCrossEntropyLoss_0"
1111
op_type: "SoftmaxCrossEntropyLoss"
12+
attribute {
13+
name: "ignore_index"
14+
i: -100
15+
type: INT
16+
}
1217
attribute {
1318
name: "reduction"
1419
s: "mean"

0 commit comments

Comments
 (0)