8.1.1.2.1.2. blueoil.converter.frontend¶
Definition of dlk.io package.
8.1.1.2.1.2.1. Submodules¶
8.1.1.2.1.2.2. Package Contents¶
8.1.1.2.1.2.2.1. Classes¶
IO class that reads/writes a model from/to TensorFlow pb. |
-
class
blueoil.converter.frontend.TensorFlowIO¶ Bases:
blueoil.converter.frontend.base.BaseIOIO 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
-