res/TensorFlowLiteRecipes/ArgMin_002/test.recipe

   1 operand {
   2   name: "ifm"
   3   type: FLOAT32
   4   shape { dim: 4 dim: 5 }
   5 }
   6 operand {
   7   name: "ofm"
   8   type: INT64
   9   shape { dim: 4 }
  10 }
  11 operand {
  12   name: "argmin/dim"
  13   type: INT32
  14   shape { }
  15   filler {
  16     tag: "explicit"
  17     arg: "1"
  18   }
  19 }
  20 operation {
  21   type: "ArgMin"
  22   argmin_options {
  23     output_type: INT64
  24   }
  25   input: "ifm"
  26   input: "argmin/dim"
  27   output: "ofm"
  28 }
  29 input: "ifm"
  30 output: "ofm"