mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
443a58e03d
Summary: Almost there, feel free to review. these c10 operators are exported to _caffe2 domain. TODO: - [x] let the onnx checker pass - [x] test tensor list as argument - [x] test caffe2 backend and converter - [x] check the c10 schema can be exported to onnx - [x] refactor the test case to share some code - [x] fix the problem in ONNX_ATEN_FALLBACK Pull Request resolved: https://github.com/pytorch/pytorch/pull/18210 Reviewed By: zrphercule Differential Revision: D14600916 Pulled By: houseroad fbshipit-source-id: 2592a75f21098fb6ceb38c5d00ee40e9e01cd144
18 lines
642 B
Plaintext
18 lines
642 B
Plaintext
ModelProto {
|
|
producer_name: "pytorch"
|
|
domain: ""
|
|
doc_string: ""
|
|
graph:
|
|
GraphProto {
|
|
name: "torch-jit-export"
|
|
inputs: [{name: "0", type:Tensor dims: 3 4},{name: "1", type:Tensor dims: 3 4}]
|
|
outputs: [{name: "3", type:Tensor dims: 3 3},{name: "4", type:Tensor dims: 3 4}]
|
|
initializers: []
|
|
nodes: [
|
|
Node {type: "Add", inputs: [0,1], outputs: [2], attributes: []},
|
|
Node {type: "ATen", inputs: [2], outputs: [3,4], attributes: [{ name: 'operator', type: string, value: 'qr'}]}
|
|
]
|
|
}
|
|
opset_import: [OperatorSetIdProto { domain: }OperatorSetIdProto { domain: org.pytorch.aten}],
|
|
}
|