mltk.core.TfliteLayer

class TfliteLayer

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)

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)

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)

Get layer input tensor as TfliteTensor

Return type:

TfliteTensor

get_input_data(index=0)

Get layer input tensor as np.ndarray

Return type:

ndarray

get_output_tensor(index=0)

Layer output tensor as TfliteTensor

Return type:

TfliteTensor

get_output_data(index=0)

Layer output tensor as np.ndarray

Return type:

ndarray