blob: 34c26f17e8b6f2ab6a747aa177172a3b7e036ca5 [file] [log] [blame]
ir_version: 8
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::ArgMax_0"
output: "1"
name: "ArgMax_0"
op_type: "ArgMax"
attribute {
name: "axis"
i: 1
type: INT
}
attribute {
name: "keepdims"
i: 0
type: INT
}
attribute {
name: "select_last_index"
i: 0
type: INT
}
}
name: "main_graph"
input {
name: "onnx::ArgMax_0"
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: 17
}