mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
141d81d095
* Move ONNX integration tests from onnx-fb-universe to PyTorch repo
* Switch to use torchvision
* Delete single rnn operator tests, they have been covered in e2e tests in test_caffe2.py
* Mirror the fix in onnx-fb-universe to bypass cuda check
667326d84b
76 lines
1.2 KiB
Text
76 lines
1.2 KiB
Text
ir_version: 3
|
|
producer_name: "pytorch"
|
|
producer_version: "0.3"
|
|
graph {
|
|
node {
|
|
input: "0"
|
|
input: "1"
|
|
output: "2"
|
|
op_type: "Add"
|
|
}
|
|
node {
|
|
input: "0"
|
|
input: "2"
|
|
output: "3"
|
|
op_type: "Mul"
|
|
}
|
|
name: "torch-jit-export"
|
|
initializer {
|
|
dims: 2
|
|
dims: 2
|
|
data_type: INT64
|
|
name: "1"
|
|
raw_data: "\001\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000\003\000\000\000\000\000\000\000\004\000\000\000\000\000\000\000"
|
|
}
|
|
input {
|
|
name: "0"
|
|
type {
|
|
tensor_type {
|
|
elem_type: INT64
|
|
shape {
|
|
dim {
|
|
dim_value: 2
|
|
}
|
|
dim {
|
|
dim_value: 2
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
input {
|
|
name: "1"
|
|
type {
|
|
tensor_type {
|
|
elem_type: INT64
|
|
shape {
|
|
dim {
|
|
dim_value: 2
|
|
}
|
|
dim {
|
|
dim_value: 2
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
output {
|
|
name: "3"
|
|
type {
|
|
tensor_type {
|
|
elem_type: INT64
|
|
shape {
|
|
dim {
|
|
dim_value: 2
|
|
}
|
|
dim {
|
|
dim_value: 2
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
opset_import {
|
|
version: 6
|
|
}
|