res/TensorFlowLiteRecipes/While_003/test.recipe

   1 version: 1
   2
   3 graph {
   4   operand {
   5     name: "cifm1"
   6     type: INT32
   7     shape { dim: 1 dim: 1 dim: 8 }
   8   }
   9   operand {
  10     name: "cifm2"
  11     type: INT32
  12     shape { }
  13     filler {
  14       tag: "explicit"
  15       arg: "10"
  16     }
  17   }
  18   operand {
  19     name: "cofm"
  20     type: BOOL
  21     shape { dim: 1 dim: 1 dim: 8 }
  22   }
  23   operation {
  24     type: "Equal"
  25     input: "cifm1"
  26     input: "cifm2"
  27     output: "cofm"
  28   }
  29   input: "cifm1"
  30   output: "cofm"
  31   name: "WHILE_COND"
  32 }
  33
  34 graph {
  35   operand {
  36     name: "bifm1"
  37     type: INT32
  38     shape { dim: 1 dim: 1 dim: 8 }
  39   }
  40   operand {
  41     name: "bifm3"
  42     type: INT32
  43     shape { }
  44     filler {
  45       tag: "explicit"
  46       arg: "1"
  47     }
  48   }
  49   operand {
  50     name: "bofm"
  51     type: INT32
  52     shape { dim: 1 dim: 1 dim: 8 }
  53   }
  54   operation {
  55     type: "Add"
  56     input: "bifm1"
  57     input: "bifm3"
  58     output: "bofm"
  59     add_options {
  60       activation: NONE
  61     }
  62   }
  63   input: "bifm1"
  64   output: "bofm"
  65   name: "WHILE_BODY"
  66 }
  67
  68 operand {
  69   name: "ifm1"
  70   type: INT32
  71   shape { dim: 1 dim: 0 dim: 8 }
  72 }
  73 operand {
  74   name: "ofm"
  75   type: INT32
  76   shape { dim: 1 dim: 1 dim: 8 }
  77 }
  78 operation {
  79   type: "While"
  80   input: "ifm1"
  81   output: "ofm"
  82   while_options {
  83     body_subgraph_index: 2
  84     cond_subgraph_index: 1
  85   }
  86 }
  87 input: "ifm1"
  88 output: "ofm"
  89 name: "Main"
  90
  91 # This is a case where main ifm1 node is valid unknown shape input, not a constant