res/TensorFlowLiteRecipes/ArgMax_001/test.recipe


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

operand {
  name: "ifm"
  type: FLOAT32
  shape { dim: 4 dim: 5 }
}
operand {
  name: "ofm"
  type: INT64
  shape { dim: 5 }
}
operand {
  name: "argmax/dim"
  type: INT32
  shape { }
  filler {
    tag: "explicit"
    arg: "0"
  }
}
operation {
  type: "ArgMax"
  argmax_options {
    output_type: INT64
  }
  input: "ifm"
  input: "argmax/dim"
  output: "ofm"
}
input: "ifm"
output: "ofm"