mirror of
https://github.com/zebrajr/pytorch.git
synced 2026-01-15 12:15:51 +00:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/21113 ghimport-source-id: 9c4ba63457a72bfc41894387e0b01be3fd9a9baf Test Plan: Imported from OSS Differential Revision: D15552204 Pulled By: izdeby fbshipit-source-id: a608213668649d058e22b510d7755cb99e7d0037
21 lines
920 B
Plaintext
21 lines
920 B
Plaintext
ModelProto {
|
|
producer_name: "pytorch"
|
|
domain: ""
|
|
doc_string: ""
|
|
graph:
|
|
GraphProto {
|
|
name: "torch-jit-export"
|
|
inputs: [{name: "0", type:Tensor dims: 3 4}]
|
|
outputs: [{name: "5", type:Tensor dims: 0}]
|
|
initializers: []
|
|
nodes: [
|
|
Node {type: "Constant", inputs: [], outputs: [1], attributes: [{ name: 'value', type: tensor, value:TensorProto shape: []}]},
|
|
Node {type: "Less", inputs: [0,1], outputs: [2], attributes: []},
|
|
Node {type: "Cast", inputs: [2], outputs: [3], attributes: [{ name: 'to', type: int, value: 2}]},
|
|
Node {type: "Cast", inputs: [3], outputs: [4], attributes: [{ name: 'to', type: int, value: 9}]},
|
|
Node {type: "ATen", inputs: [0,4], outputs: [5], attributes: [{ name: 'operator', type: string, value: 'index'}]}
|
|
]
|
|
}
|
|
opset_import: [OperatorSetIdProto { domain: }OperatorSetIdProto { domain: org.pytorch.aten}],
|
|
}
|