mltk.core.TfliteLayer

class TfliteLayer[source]

Wrapper for TFLite flatbuffer layer

Properties

index

Index of this layer in the model

inputs

List of layer input tensor(s)

metadata

Additional key/value data to associated with layer

model

Reference to associated TfliteModel

n_inputs

Return the number of inputs

n_outputs

Return the number of outputs

name

op<index>-<OpCodeStr>

opcode

OpCode numeric value

opcode_str

OpCode as a string

options

Layer-specific options/config

outputs

List of layer output tensor(s)

Methods

__init__

from_flatbuffer

Instantiate a TfliteLayer from then given TfliteModel flatbuffer operation

get_input_data

Get layer input tensor as np.ndarray

get_input_tensor

Get layer input tensor as TfliteTensor

get_output_data

Layer output tensor as np.ndarray

get_output_tensor

Layer output tensor as TfliteTensor

static from_flatbuffer(index, model, fb_operation)[source]

Instantiate a TfliteLayer from then given TfliteModel flatbuffer operation

Return type:

TfliteLayer

Parameters:
  • index (int) –

  • model (TfliteModel) –

  • fb_operation (OperatorT) –

__init__(index, opcode, opcode_version, model, fb_operation)[source]
Parameters:
  • index (int) –

  • opcode (BuiltinOperator) –

  • opcode_version (int) –

  • model (TfliteModel) –

  • fb_operation (OperatorT) –

property index: int

Index of this layer in the model

Return type:

int

property name: str

op<index>-<OpCodeStr>

Type:

Name of current layer as

Return type:

str

property opcode: BuiltinOperator

OpCode numeric value

Return type:

BuiltinOperator

property opcode_str: str

OpCode as a string

Return type:

str

property options: TfliteLayerOptions

Layer-specific options/config

Return type:

TfliteLayerOptions

property model: TfliteModel

Reference to associated TfliteModel

Return type:

TypeVar(TfliteModel)

property metadata: Dict[str, object]

Additional key/value data to associated with layer

NOTE: This information is generated by the framework/Python scripts

(i.e. The information does NOT come from the .tflite model)

Return type:

Dict[str, object]

property inputs: List[TfliteTensor]

List of layer input tensor(s)

Return type:

List[TfliteTensor]

property n_inputs: int

Return the number of inputs

Return type:

int

property outputs: List[TfliteTensor]

List of layer output tensor(s)

Return type:

List[TfliteTensor]

property n_outputs: int

Return the number of outputs

Return type:

int

get_input_tensor(index=0)[source]

Get layer input tensor as TfliteTensor

Return type:

TfliteTensor

get_input_data(index=0)[source]

Get layer input tensor as np.ndarray

Return type:

ndarray

get_output_tensor(index=0)[source]

Layer output tensor as TfliteTensor

Return type:

TfliteTensor

get_output_data(index=0)[source]

Layer output tensor as np.ndarray

Return type:

ndarray