Files
pytorch/test/onnx/expect/TestOperators.test_argmax.expect
Lara Haidar-Ahmad 073634612f ONNX Export Argmin and Argmax ops
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17382

Differential Revision: D14338811

Pulled By: houseroad

fbshipit-source-id: be07548d8063d1aa94f1801c18137738365b85fb
2019-03-06 12:11:47 -08:00

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
}