mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/20116 Differential Revision: D15256899 Pulled By: houseroad fbshipit-source-id: cf0dfd679d528fbb77f483e23071f4a96fb27091
273 lines
3.9 KiB
Plaintext
273 lines
3.9 KiB
Plaintext
ir_version: 4
|
|
producer_name: "pytorch"
|
|
producer_version: "1.1"
|
|
graph {
|
|
node {
|
|
output: "1"
|
|
op_type: "Constant"
|
|
attribute {
|
|
name: "value"
|
|
t {
|
|
data_type: 7
|
|
raw_data: "\002\000\000\000\000\000\000\000"
|
|
}
|
|
type: TENSOR
|
|
}
|
|
}
|
|
node {
|
|
input: "input"
|
|
output: "2"
|
|
op_type: "Shape"
|
|
}
|
|
node {
|
|
input: "2"
|
|
input: "1"
|
|
output: "3"
|
|
op_type: "Gather"
|
|
attribute {
|
|
name: "axis"
|
|
i: 0
|
|
type: INT
|
|
}
|
|
}
|
|
node {
|
|
output: "4"
|
|
op_type: "Constant"
|
|
attribute {
|
|
name: "value"
|
|
t {
|
|
data_type: 7
|
|
raw_data: "\002\000\000\000\000\000\000\000"
|
|
}
|
|
type: TENSOR
|
|
}
|
|
}
|
|
node {
|
|
input: "3"
|
|
input: "4"
|
|
output: "5"
|
|
op_type: "Mul"
|
|
}
|
|
node {
|
|
input: "5"
|
|
output: "6"
|
|
op_type: "Floor"
|
|
}
|
|
node {
|
|
output: "7"
|
|
op_type: "Constant"
|
|
attribute {
|
|
name: "value"
|
|
t {
|
|
data_type: 7
|
|
raw_data: "\003\000\000\000\000\000\000\000"
|
|
}
|
|
type: TENSOR
|
|
}
|
|
}
|
|
node {
|
|
input: "input"
|
|
output: "8"
|
|
op_type: "Shape"
|
|
}
|
|
node {
|
|
input: "8"
|
|
input: "7"
|
|
output: "9"
|
|
op_type: "Gather"
|
|
attribute {
|
|
name: "axis"
|
|
i: 0
|
|
type: INT
|
|
}
|
|
}
|
|
node {
|
|
output: "10"
|
|
op_type: "Constant"
|
|
attribute {
|
|
name: "value"
|
|
t {
|
|
data_type: 7
|
|
raw_data: "\002\000\000\000\000\000\000\000"
|
|
}
|
|
type: TENSOR
|
|
}
|
|
}
|
|
node {
|
|
input: "9"
|
|
input: "10"
|
|
output: "11"
|
|
op_type: "Mul"
|
|
}
|
|
node {
|
|
input: "11"
|
|
output: "12"
|
|
op_type: "Floor"
|
|
}
|
|
node {
|
|
input: "6"
|
|
output: "13"
|
|
op_type: "Unsqueeze"
|
|
attribute {
|
|
name: "axes"
|
|
ints: 0
|
|
type: INTS
|
|
}
|
|
}
|
|
node {
|
|
input: "12"
|
|
output: "14"
|
|
op_type: "Unsqueeze"
|
|
attribute {
|
|
name: "axes"
|
|
ints: 0
|
|
type: INTS
|
|
}
|
|
}
|
|
node {
|
|
input: "13"
|
|
input: "14"
|
|
output: "15"
|
|
op_type: "Concat"
|
|
attribute {
|
|
name: "axis"
|
|
i: 0
|
|
type: INT
|
|
}
|
|
}
|
|
node {
|
|
output: "16"
|
|
op_type: "Constant"
|
|
attribute {
|
|
name: "value"
|
|
t {
|
|
dims: 2
|
|
data_type: 1
|
|
raw_data: "\000\000\200?\000\000\200?"
|
|
}
|
|
type: TENSOR
|
|
}
|
|
}
|
|
node {
|
|
input: "15"
|
|
output: "17"
|
|
op_type: "Cast"
|
|
attribute {
|
|
name: "to"
|
|
i: 1
|
|
type: INT
|
|
}
|
|
}
|
|
node {
|
|
input: "input"
|
|
output: "18"
|
|
op_type: "Shape"
|
|
}
|
|
node {
|
|
input: "18"
|
|
output: "19"
|
|
op_type: "Slice"
|
|
attribute {
|
|
name: "axes"
|
|
ints: 0
|
|
type: INTS
|
|
}
|
|
attribute {
|
|
name: "ends"
|
|
ints: 4
|
|
type: INTS
|
|
}
|
|
attribute {
|
|
name: "starts"
|
|
ints: 2
|
|
type: INTS
|
|
}
|
|
}
|
|
node {
|
|
input: "19"
|
|
output: "20"
|
|
op_type: "Cast"
|
|
attribute {
|
|
name: "to"
|
|
i: 1
|
|
type: INT
|
|
}
|
|
}
|
|
node {
|
|
input: "17"
|
|
input: "20"
|
|
output: "21"
|
|
op_type: "Div"
|
|
}
|
|
node {
|
|
input: "16"
|
|
input: "21"
|
|
output: "22"
|
|
op_type: "Concat"
|
|
attribute {
|
|
name: "axis"
|
|
i: 0
|
|
type: INT
|
|
}
|
|
}
|
|
node {
|
|
input: "input"
|
|
input: "22"
|
|
output: "23"
|
|
op_type: "Upsample"
|
|
attribute {
|
|
name: "mode"
|
|
s: "nearest"
|
|
type: STRING
|
|
}
|
|
}
|
|
name: "torch-jit-export"
|
|
input {
|
|
name: "input"
|
|
type {
|
|
tensor_type {
|
|
elem_type: 1
|
|
shape {
|
|
dim {
|
|
dim_value: 1
|
|
}
|
|
dim {
|
|
dim_value: 2
|
|
}
|
|
dim {
|
|
dim_value: 3
|
|
}
|
|
dim {
|
|
dim_value: 4
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
output {
|
|
name: "23"
|
|
type {
|
|
tensor_type {
|
|
elem_type: 1
|
|
shape {
|
|
dim {
|
|
dim_value: 1
|
|
}
|
|
dim {
|
|
dim_value: 2
|
|
}
|
|
dim {
|
|
dim_value: 6
|
|
}
|
|
dim {
|
|
dim_value: 8
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
opset_import {
|
|
version: 9
|
|
}
|