res/TensorFlowLiteRecipes/OneHot_002/test.recipe

   1 operand {
   2   name: "indices"
   3   type: INT32
   4   shape { dim: 2 dim: 2 }
   5 }
   6 operand {
   7   name: "depth"
   8   type: INT32
   9   shape {}
  10   filler { tag: "explicit" arg: "3" }
  11 }
  12 operand {
  13   name: "on_value"
  14   type: FLOAT32
  15   shape {}
  16 }
  17 operand {
  18   name: "off_value"
  19   type: FLOAT32
  20   shape {}
  21 }
  22 operand {
  23   name: "ofm"
  24   type: FLOAT32
  25   shape { dim: 2 dim: 2 dim: 3 }
  26 }
  27 operation {
  28   type: "OneHot"
  29   onehot_options {
  30     axis: -1
  31   }
  32   input: "indices"
  33   input: "depth"
  34   input: "on_value"
  35   input: "off_value"
  36   output: "ofm"
  37 }
  38 input: "indices"
  39 input: "on_value"
  40 input: "off_value"
  41 output: "ofm"