mirror of https://github.com/explosion/spaCy.git
a83a501195
The `forward` of `precomputable_biaffine` performs matrix multiplication and then `vstack`s the result with padding. This creates a temporary array used for the output of matrix concatenation. This change avoids the temporary by pre-allocating an array that is large enough for the output of matrix multiplication plus padding and fills the array in-place. This gave me a small speedup (a bit over 100 WPS) on de_core_news_lg on M1 Max (after changing thinc-apple-ops to support in-place gemm as BLIS does). |
||
|---|---|---|
| .. | ||
| models | ||
| __init__.py | ||
| _character_embed.py | ||
| _precomputable_affine.py | ||
| callbacks.py | ||
| extract_ngrams.py | ||
| extract_spans.py | ||
| featureextractor.py | ||
| parser_model.pxd | ||
| parser_model.pyx | ||
| staticvectors.py | ||
| tb_framework.py | ||