8.1.1.2.1.2.1.2. blueoil.converter.frontend.tensorflow

Module of TensorFlow IO.

8.1.1.2.1.2.1.2.1. Module Contents

8.1.1.2.1.2.1.2.1.1. Classes

TensorFlowIO

IO class that reads/writes a model from/to TensorFlow pb.

class blueoil.converter.frontend.tensorflow.TensorFlowIO

Bases: blueoil.converter.frontend.base.BaseIO

IO class that reads/writes a model from/to TensorFlow pb.

read(self, pb_path: str) → Graph

Read TF file and load model.

Parameters

pb_path (str) – Path to TF file

Returns

Loaded model

Return type

Model

abstract write(self, graph: Graph, path: str) → None

Write the model to a file.

Parameters

model (Model) – Model to be written