compiler/tflchef/tests/no_shape/test.recipe

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