ngraph/test/models/onnx/loop_2d_add_no_identity_cond.prototxt

   1 ir_version: 6
   2 producer_name: "nGraph ONNX Importer"
   3 graph {
   4   name: "basic loop"
   5   node {
   6     input: "trip_count"
   7     input: ""
   8     input: "a_init"
   9     output: "a_final"
  10     output: "a_values"
  11     op_type: "Loop"
  12     attribute {
  13       name: "body"
  14       g {
  15         node {
  16           input: "a_in"
  17           input: "b"
  18           output: "current_a"
  19           name: "loop_body_add"
  20           op_type: "Add"
  21         }
  22         node {
  23           input: "cond"
  24           output: "cond_out"
  25           name: "cond_identity"
  26           op_type: "Not"
  27         }
  28         node {
  29           input: "current_a"
  30           output: "a_out"
  31           name: "output_accumulator"
  32           op_type: "Identity"
  33         }
  34         name: "simple add"
  35         initializer {
  36           dims: 1
  37           dims: 2
  38           data_type: 1
  39           float_data: 1
  40           float_data: 1
  41           name: "b"
  42         }
  43         input {
  44           name: "i"
  45           type {
  46             tensor_type {
  47               elem_type: 7
  48               shape {
  49                 dim {
  50                   dim_value: 1
  51                 }
  52               }
  53             }
  54           }
  55         }
  56         input {
  57           name: "cond"
  58           type {
  59             tensor_type {
  60               elem_type: 9
  61               shape {
  62               }
  63             }
  64           }
  65         }
  66         input {
  67           name: "a_in"
  68           type {
  69             tensor_type {
  70               elem_type: 1
  71               shape {
  72                 dim {
  73                   dim_value: 1
  74                 }
  75                 dim {
  76                   dim_value: 2
  77                 }
  78               }
  79             }
  80           }
  81         }
  82         output {
  83           name: "cond_out"
  84           type {
  85             tensor_type {
  86               elem_type: 9
  87               shape {
  88               }
  89             }
  90           }
  91         }
  92         output {
  93           name: "current_a"
  94           type {
  95             tensor_type {
  96               elem_type: 1
  97               shape {
  98                 dim {
  99                   dim_value: 1
 100                 }
 101                 dim {
 102                   dim_value: 2
 103                 }
 104               }
 105             }
 106           }
 107         }
 108         output {
 109           name: "a_out"
 110           type {
 111             tensor_type {
 112               elem_type: 1
 113               shape {
 114                 dim {
 115                   dim_value: 1
 116                 }
 117                 dim {
 118                   dim_value: 2
 119                 }
 120               }
 121             }
 122           }
 123         }
 124       }
 125       type: GRAPH
 126     }
 127   }
 128   initializer {
 129     dims: 1
 130     data_type: 7
 131     int64_data: 3
 132     name: "trip_count"
 133   }
 134   input {
 135     name: "a_init"
 136     type {
 137       tensor_type {
 138         elem_type: 1
 139         shape {
 140           dim {
 141             dim_value: 1
 142           }
 143           dim {
 144             dim_value: 2
 145           }
 146         }
 147       }
 148     }
 149   }
 150   output {
 151     name: "a_final"
 152     type {
 153       tensor_type {
 154         elem_type: 1
 155         shape {
 156           dim {
 157             dim_value: 1
 158           }
 159           dim {
 160             dim_value: 2
 161           }
 162         }
 163       }
 164     }
 165   }
 166   output {
 167     name: "a_values"
 168     type {
 169       tensor_type {
 170         elem_type: 1
 171         shape {
 172           dim {
 173             dim_value: 3
 174           }
 175           dim {
 176             dim_value: 2
 177           }
 178         }
 179       }
 180     }
 181   }
 182 }
 183 opset_import {
 184   version: 11
 185 }