blob: 2ef292045834d30d292224aa5549021b6a9ebb62 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
operand {
name: "ifm"
type: UINT8
shape { dim: 4 dim: 5 dim: 6 }
quant { min: 0 max: 255 scale: 1.0 zero_point: 0 }
}
operand {
name: "ofm"
type: INT64
shape { dim: 4 dim: 6 }
}
operand {
name: "argmax/dim"
type: INT32
shape { }
filler {
tag: "explicit"
arg: "1"
}
}
operation {
type: "ArgMax"
argmax_options {
output_type: INT64
}
input: "ifm"
input: "argmax/dim"
output: "ofm"
}
input: "ifm"
output: "ofm"
|