mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
6fb79b7004
Except for the usual location, had to update the version in one of ONNX expect patterns, namely here: 43660051d8/test/onnx/expect/TestOperators.test_avg_pool2d.expect (L3)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/90491
Approved by: https://github.com/jansel, https://github.com/albanD
106 lines
1.9 KiB
Plaintext
106 lines
1.9 KiB
Plaintext
ir_version: 7
|
|
producer_name: "pytorch"
|
|
producer_version: "2.0.0"
|
|
graph {
|
|
node {
|
|
output: "onnx::Pad_1"
|
|
name: "Constant_0"
|
|
op_type: "Constant"
|
|
attribute {
|
|
name: "value"
|
|
t {
|
|
dims: 8
|
|
data_type: 7
|
|
raw_data: "\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000"
|
|
}
|
|
type: TENSOR
|
|
}
|
|
}
|
|
node {
|
|
input: "onnx::Pad_0"
|
|
input: "onnx::Pad_1"
|
|
output: "onnx::AveragePool_2"
|
|
name: "Pad_1"
|
|
op_type: "Pad"
|
|
attribute {
|
|
name: "mode"
|
|
s: "constant"
|
|
type: STRING
|
|
}
|
|
}
|
|
node {
|
|
input: "onnx::AveragePool_2"
|
|
output: "3"
|
|
name: "AveragePool_2"
|
|
op_type: "AveragePool"
|
|
attribute {
|
|
name: "kernel_shape"
|
|
ints: 3
|
|
ints: 3
|
|
type: INTS
|
|
}
|
|
attribute {
|
|
name: "pads"
|
|
ints: 0
|
|
ints: 0
|
|
ints: 0
|
|
ints: 0
|
|
type: INTS
|
|
}
|
|
attribute {
|
|
name: "strides"
|
|
ints: 2
|
|
ints: 2
|
|
type: INTS
|
|
}
|
|
}
|
|
name: "torch_jit"
|
|
input {
|
|
name: "onnx::Pad_0"
|
|
type {
|
|
tensor_type {
|
|
elem_type: 1
|
|
shape {
|
|
dim {
|
|
dim_value: 20
|
|
}
|
|
dim {
|
|
dim_value: 16
|
|
}
|
|
dim {
|
|
dim_value: 50
|
|
}
|
|
dim {
|
|
dim_value: 32
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
output {
|
|
name: "3"
|
|
type {
|
|
tensor_type {
|
|
elem_type: 1
|
|
shape {
|
|
dim {
|
|
dim_value: 20
|
|
}
|
|
dim {
|
|
dim_value: 16
|
|
}
|
|
dim {
|
|
dim_value: 24
|
|
}
|
|
dim {
|
|
dim_value: 15
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
opset_import {
|
|
version: 14
|
|
}
|