• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1ModelProto {
2  producer_name: "pytorch"
3  domain: ""
4  doc_string: ""
5  graph:
6    GraphProto {
7      name: "torch-jit-export"
8      inputs: [{name: "0", type:Tensor dims: 3 4}]
9      outputs: [{name: "5", type:Tensor dims: 0}]
10      initializers: []
11      nodes: [
12        Node {type: "Constant", inputs: [], outputs: [1], attributes: [{ name: 'value', type: tensor, value:TensorProto shape: []}]},
13        Node {type: "Less", inputs: [0,1], outputs: [2], attributes: []},
14        Node {type: "Cast", inputs: [2], outputs: [3], attributes: [{ name: 'to', type: int, value: 2}]},
15        Node {type: "Cast", inputs: [3], outputs: [4], attributes: [{ name: 'to', type: int, value: 9}]},
16        Node {type: "ATen", domain: "org.pytorch.aten", inputs: [0,4], outputs: [5], attributes: [{ name: 'operator', type: string, value: 'index'}, { name: 'overload_name', type: string, value: ''}]}
17      ]
18    }
19  opset_import: [OperatorSetIdProto { domain: }OperatorSetIdProto { domain: org.pytorch.aten}],
20}
21