Computing LSTM forward layer without allocating (#3351)

unicode-org · May 4, 2023 · 5ab0f5f · 5ab0f5f
1 parent 2f5c923
commit 5ab0f5f
Show file tree

Hide file tree

Showing 4 changed files with 213 additions and 184 deletions.
diff --git a/components/segmenter/src/complex/lstm/matrix.rs b/components/segmenter/src/complex/lstm/matrix.rs
@@ -211,6 +211,7 @@ impl<'a, const D: usize> MatrixBorrowedMut<'a, D> {
         Some(())
     }
 
+    #[allow(dead_code)] // maybe needed for more complicated bies calculations
     /// Mutates this matrix by applying a softmax transformation.
     pub(super) fn softmax_transform(&mut self) {
         for v in self.data.iter_mut() {