feat: add linear layer to lstm network

BREAKING CHANGES

.pre-commit-config.yaml

@@ -22,7 +22,7 @@ repos:
     -   id: no-commit-to-branch
         stages: [ commit, manual ]
 -   repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.0.1
+    rev: v1.1.1
     hooks:
     -   id: mypy
         stages: [ manual ]

elasticai/creator/nn/linear.py

@@ -32,9 +32,6 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
         return self.ops.matmul(x, weight.T)
 
-    def quantized_forward(self, x: torch.Tensor) -> torch.Tensor:
-        raise NotImplementedError("The quantized_forward function is not implemented.")
-
 
 @dataclass
 class FixedPointConfig:

elasticai/creator/on_disk_path.py

@@ -0,0 +1,29 @@
+import os
+from pathlib import Path as _PyPath
+from typing import Iterable, Optional
+
+from elasticai.creator.hdl.translatable import File, Path
+
+
+class OnDiskFile(File):
+    def __init__(self, full_path: str):
+        self._full_path = full_path
+
+    def write_text(self, text: Iterable[str]) -> None:
+        full_path = _PyPath(self._full_path)
+        folder = full_path.parent
+        if not folder.exists():
+            os.makedirs(folder)
+        with open(full_path, "w") as f:
+            f.writelines((f"{line}\n" for line in text))
+
+
+class OnDiskPath(Path):
+    def __init__(self, name: str, parent: str = "."):
+        self._full_path = f"{parent}/{name}"
+
+    def create_subpath(self, name: str) -> "Path":
+        return OnDiskPath(name, parent=self._full_path)
+
+    def as_file(self, suffix: str) -> "File":
+        return OnDiskFile(full_path=f"{self._full_path}{suffix}")