blob: b0dc5d03c02462dbe25f82b7fe6272bc5dfbc54b [file] [log] [blame]
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
output: "onnx::Where_1"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 3
data_type: 7
raw_data: "\004\000\000\000\000\000\000\000\006\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::ConstantOfShape_10"
name: "Constant_1"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\003\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::ConstantOfShape_10"
output: "onnx::Mul_3"
name: "ConstantOfShape_2"
op_type: "ConstantOfShape"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Mul_4"
name: "Constant_3"
op_type: "Constant"
attribute {
name: "value"
t {
data_type: 7
raw_data: "\377\377\377\377\377\377\377\377"
}
type: TENSOR
}
}
node {
input: "onnx::Mul_3"
input: "onnx::Mul_4"
output: "onnx::Equal_5"
name: "Mul_4"
op_type: "Mul"
}
node {
output: "onnx::Equal_6"
name: "Constant_5"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 3
data_type: 7
raw_data: "\004\000\000\000\000\000\000\000\006\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Equal_6"
input: "onnx::Equal_5"
output: "onnx::Where_7"
name: "Equal_6"
op_type: "Equal"
}
node {
input: "onnx::Where_7"
input: "onnx::Mul_3"
input: "onnx::Where_1"
output: "onnx::Expand_8"
name: "Where_7"
op_type: "Where"
}
node {
input: "onnx::Expand_0"
input: "onnx::Expand_8"
output: "9"
name: "Expand_8"
op_type: "Expand"
}
name: "torch_jit"
input {
name: "onnx::Expand_0"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 6
}
dim {
dim_value: 1
}
}
}
}
}
output {
name: "9"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 4
}
dim {
dim_value: 6
}
dim {
dim_value: 2
}
}
}
}
}
}
opset_import {
version: 14
}