forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathTestScript.test_onnx_export_speculate-f2.expect
65 lines (61 loc) · 2.97 KB
/
TestScript.test_onnx_export_speculate-f2.expect
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
ModelProto {
producer_name: "pytorch"
domain: ""
doc_string: ""
graph:
GraphProto {
name: "torch-jit-export"
inputs: [{name: "x.1", type:Tensor dims: 1 10},{name: "1", type:Tensor dims: 20 10},{name: "2", type:Tensor dims: 20}]
outputs: [{name: "8", type:Tensor dims: 1 20}]
initializers: [TensorProto shape: [20 10],TensorProto shape: [20]]
nodes: [
Node {type: "Add", inputs: [x.1,x.1], outputs: [3], attributes: []},
Node {type: "ReduceSum", inputs: [3], outputs: [4], attributes: [{ name: 'keepdims', type: int, value: 0}]},
Node {type: "Constant", inputs: [], outputs: [5], attributes: [{ name: 'value', type: tensor, value:TensorProto shape: []}]},
Node {type: "Greater", inputs: [4,5], outputs: [6], attributes: []},
Node {type: "Cast", inputs: [6], outputs: [7], attributes: [{ name: 'to', type: int, value: 2}]},
Node {type: "If", inputs: [7], outputs: [8], attributes: [{ name: 'then_branch', type: graph, value:
GraphProto {
name: "torch-jit-export1"
inputs: []
outputs: [{name: "9", type:Tensor dims: 1 20}]
initializers: []
nodes: [
Node {type: "If", inputs: [7], outputs: [9], attributes: [{ name: 'then_branch', type: graph, value:
GraphProto {
name: "torch-jit-export2"
inputs: []
outputs: [{name: "10", type:Tensor dims: 1 20}]
initializers: []
nodes: [
Node {type: "Gemm", inputs: [3,1,2], outputs: [10], attributes: [{ name: 'alpha', type: float, value: 1},{ name: 'beta', type: float, value: 1},{ name: 'transB', type: int, value: 1}]}
]
}
},{ name: 'else_branch', type: graph, value:
GraphProto {
name: "torch-jit-export3"
inputs: []
outputs: [{name: "11", type:Tensor dims: 1 20}]
initializers: []
nodes: [
Node {type: "Gemm", inputs: [3,1,2], outputs: [11], attributes: [{ name: 'alpha', type: float, value: 1},{ name: 'beta', type: float, value: 1},{ name: 'transB', type: int, value: 1}]}
]
}
}]}
]
}
},{ name: 'else_branch', type: graph, value:
GraphProto {
name: "torch-jit-export4"
inputs: []
outputs: [{name: "12", type:Tensor dims: 1 20}]
initializers: []
nodes: [
Node {type: "Gemm", inputs: [3,1,2], outputs: [12], attributes: [{ name: 'alpha', type: float, value: 1},{ name: 'beta', type: float, value: 1},{ name: 'transB', type: int, value: 1}]}
]
}
}]}
]
}
opset_import: [OperatorSetIdProto { domain: }],
}