tools/tflitefile_tool/tflite/SubGraph.py

   1 # automatically generated by the FlatBuffers compiler, do not modify
   2
   3 # namespace: onert_tflite
   4
   5 import flatbuffers
   6
   7
   8 class SubGraph(object):
   9     __slots__ = ['_tab']
  10
  11     @classmethod
  12     def GetRootAsSubGraph(cls, buf, offset):
  13         n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
  14         x = SubGraph()
  15         x.Init(buf, n + offset)
  16         return x
  17
  18     # SubGraph
  19     def Init(self, buf, pos):
  20         self._tab = flatbuffers.table.Table(buf, pos)
  21
  22     # SubGraph
  23     def Tensors(self, j):
  24         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
  25         if o != 0:
  26             x = self._tab.Vector(o)
  27             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
  28             x = self._tab.Indirect(x)
  29             from .Tensor import Tensor
  30             obj = Tensor()
  31             obj.Init(self._tab.Bytes, x)
  32             return obj
  33         return None
  34
  35     # SubGraph
  36     def TensorsLength(self):
  37         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
  38         if o != 0:
  39             return self._tab.VectorLen(o)
  40         return 0
  41
  42     # SubGraph
  43     def Inputs(self, j):
  44         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
  45         if o != 0:
  46             a = self._tab.Vector(o)
  47             return self._tab.Get(
  48                 flatbuffers.number_types.Int32Flags,
  49                 a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
  50         return 0
  51
  52     # SubGraph
  53     def InputsAsNumpy(self):
  54         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
  55         if o != 0:
  56             return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
  57         return 0
  58
  59     # SubGraph
  60     def InputsLength(self):
  61         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
  62         if o != 0:
  63             return self._tab.VectorLen(o)
  64         return 0
  65
  66     # SubGraph
  67     def Outputs(self, j):
  68         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
  69         if o != 0:
  70             a = self._tab.Vector(o)
  71             return self._tab.Get(
  72                 flatbuffers.number_types.Int32Flags,
  73                 a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
  74         return 0
  75
  76     # SubGraph
  77     def OutputsAsNumpy(self):
  78         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
  79         if o != 0:
  80             return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
  81         return 0
  82
  83     # SubGraph
  84     def OutputsLength(self):
  85         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
  86         if o != 0:
  87             return self._tab.VectorLen(o)
  88         return 0
  89
  90     # SubGraph
  91     def Operators(self, j):
  92         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
  93         if o != 0:
  94             x = self._tab.Vector(o)
  95             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
  96             x = self._tab.Indirect(x)
  97             from .Operator import Operator
  98             obj = Operator()
  99             obj.Init(self._tab.Bytes, x)
 100             return obj
 101         return None
 102
 103     # SubGraph
 104     def OperatorsLength(self):
 105         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
 106         if o != 0:
 107             return self._tab.VectorLen(o)
 108         return 0
 109
 110     # SubGraph
 111     def Name(self):
 112         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
 113         if o != 0:
 114             return self._tab.String(o + self._tab.Pos)
 115         return None
 116
 117
 118 def SubGraphStart(builder):
 119     builder.StartObject(5)
 120
 121
 122 def SubGraphAddTensors(builder, tensors):
 123     builder.PrependUOffsetTRelativeSlot(
 124         0, flatbuffers.number_types.UOffsetTFlags.py_type(tensors), 0)
 125
 126
 127 def SubGraphStartTensorsVector(builder, numElems):
 128     return builder.StartVector(4, numElems, 4)
 129
 130
 131 def SubGraphAddInputs(builder, inputs):
 132     builder.PrependUOffsetTRelativeSlot(
 133         1, flatbuffers.number_types.UOffsetTFlags.py_type(inputs), 0)
 134
 135
 136 def SubGraphStartInputsVector(builder, numElems):
 137     return builder.StartVector(4, numElems, 4)
 138
 139
 140 def SubGraphAddOutputs(builder, outputs):
 141     builder.PrependUOffsetTRelativeSlot(
 142         2, flatbuffers.number_types.UOffsetTFlags.py_type(outputs), 0)
 143
 144
 145 def SubGraphStartOutputsVector(builder, numElems):
 146     return builder.StartVector(4, numElems, 4)
 147
 148
 149 def SubGraphAddOperators(builder, operators):
 150     builder.PrependUOffsetTRelativeSlot(
 151         3, flatbuffers.number_types.UOffsetTFlags.py_type(operators), 0)
 152
 153
 154 def SubGraphStartOperatorsVector(builder, numElems):
 155     return builder.StartVector(4, numElems, 4)
 156
 157
 158 def SubGraphAddName(builder, name):
 159     builder.PrependUOffsetTRelativeSlot(
 160         4, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
 161
 162
 163 def SubGraphEnd(builder):
 164     return builder.EndObject()