mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17382 Differential Revision: D14338811 Pulled By: houseroad fbshipit-source-id: be07548d8063d1aa94f1801c18137738365b85fb
54 lines
737 B
Plaintext
54 lines
737 B
Plaintext
ir_version: 4
|
|
producer_name: "pytorch"
|
|
producer_version: "1.1"
|
|
graph {
|
|
node {
|
|
input: "x"
|
|
output: "1"
|
|
op_type: "ArgMax"
|
|
attribute {
|
|
name: "axis"
|
|
i: 1
|
|
type: INT
|
|
}
|
|
attribute {
|
|
name: "keepdims"
|
|
i: 0
|
|
type: INT
|
|
}
|
|
}
|
|
name: "torch-jit-export"
|
|
input {
|
|
name: "x"
|
|
type {
|
|
tensor_type {
|
|
elem_type: 1
|
|
shape {
|
|
dim {
|
|
dim_value: 4
|
|
}
|
|
dim {
|
|
dim_value: 4
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
output {
|
|
name: "1"
|
|
type {
|
|
tensor_type {
|
|
elem_type: 7
|
|
shape {
|
|
dim {
|
|
dim_value: 4
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
opset_import {
|
|
version: 10
|
|
}
|