thinc 9.0.0.dev4__tar.gz → 9.0.0.dev5__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {thinc-9.0.0.dev4/thinc.egg-info → thinc-9.0.0.dev5}/PKG-INFO +1 -1
- thinc-9.0.0.dev5/thinc/about.py +2 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/cupy_ops.py +2 -2
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/compat.py +3 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/torchscript.py +1 -1
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_basic_tagger.py +0 -1
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5/thinc.egg-info}/PKG-INFO +1 -1
- thinc-9.0.0.dev4/thinc/about.py +0 -2
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/LICENSE +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/MANIFEST.in +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/README.md +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/pyproject.toml +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/setup.cfg +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/setup.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/__init__.pxd +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/api.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/__init__.pxd +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_cupy_allocators.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_custom_kernels.cu +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_custom_kernels.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_murmur3.cu +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_param_server.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/cblas.pxd +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/cblas.pyx +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/cpu_kernels.hh +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/mps_ops.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/numpy_ops.pxd +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/numpy_ops.pyx +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/ops.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/config.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/initializers.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/add.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/array_getitem.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/bidirectional.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/cauchysimilarity.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/chain.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/clipped_linear.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/clone.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/concatenate.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/dish.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/dropout.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/embed.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/expand_window.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/gelu.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/hard_swish.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/hard_swish_mobilenet.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/hashembed.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/layernorm.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/linear.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/list2array.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/list2padded.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/list2ragged.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/logistic.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/lstm.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/map_list.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/maxout.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/mish.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/multisoftmax.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/mxnetwrapper.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/noop.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/padded2list.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/parametricattention.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/parametricattention_v2.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/premap_ids.pyx +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/pytorchwrapper.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/ragged2list.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_first.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_last.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_max.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_mean.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_sum.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/relu.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/remap_ids.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/residual.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/resizable.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/siamese.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/sigmoid.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/sigmoid_activation.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/softmax.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/softmax_activation.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/sparselinear.pyx +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/strings2arrays.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/swish.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/tensorflowwrapper.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/torchscriptwrapper.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/tuplify.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/uniqued.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_array.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_array2d.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_cpu.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_debug.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_flatten.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_flatten_v2.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_getitem.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_list.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_nvtx_range.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_padded.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_ragged.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_reshape.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_signpost_interval.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/loss.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/model.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/mypy.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/optimizers.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/py.typed +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/schedules.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/mxnet.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/pytorch.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/pytorch_grad_scaler.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/shim.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/tensorflow.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/backends/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/backends/test_mem.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/backends/test_ops.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/conftest.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/enable_mxnet.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/enable_tensorflow.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_combinators.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_feed_forward.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_hash_embed.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_layers_api.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_linear.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_lstm.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_mappers.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_mnist.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_mxnet_wrapper.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_parametric_attention_v2.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_pytorch_wrapper.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_reduce.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_resizable.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_shim.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_softmax.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_sparse_linear.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_tensorflow_wrapper.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_torchscriptwrapper.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_transforms.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_uniqued.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_with_debug.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_with_flatten.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_with_transforms.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/model/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/model/test_model.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/model/test_validation.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/configs/mypy-default.ini +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/configs/mypy-plugin.ini +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/fail_no_plugin.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/fail_plugin.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/success_no_plugin.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/success_plugin.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/outputs/fail-no-plugin.txt +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/outputs/fail-plugin.txt +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/outputs/success-no-plugin.txt +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/outputs/success-plugin.txt +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/test_mypy.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/issue519/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/issue519/program.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/issue519/test_issue519.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/test_issue208.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/test_issue564.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/shims/__init__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/shims/test_pytorch_grad_scaler.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/strategies.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_config.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_examples.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_import__all__.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_indexing.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_initializers.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_loss.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_optimizers.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_schedules.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_serialize.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_types.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_util.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/util.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/types.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/util.py +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/SOURCES.txt +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/dependency_links.txt +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/entry_points.txt +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/not-zip-safe +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/requires.txt +0 -0
- {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/top_level.txt +0 -0
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import numpy
|
|
2
2
|
|
|
3
3
|
from .. import registry
|
|
4
|
-
from ..compat import cupy, cupyx
|
|
4
|
+
from ..compat import cublas, cupy, cupyx
|
|
5
5
|
from ..types import DeviceTypes
|
|
6
6
|
from ..util import (
|
|
7
7
|
is_cupy_array,
|
|
@@ -257,7 +257,7 @@ class CupyOps(Ops):
|
|
|
257
257
|
# implementation.
|
|
258
258
|
def frobenius_norm(X):
|
|
259
259
|
X_vec = X.reshape(-1)
|
|
260
|
-
return
|
|
260
|
+
return cublas.nrm2(X_vec)
|
|
261
261
|
|
|
262
262
|
grad_norm = cupy.maximum(frobenius_norm(gradient), 1e-12)
|
|
263
263
|
gradient *= cupy.minimum(threshold, grad_norm) / grad_norm
|
|
@@ -4,9 +4,11 @@ from packaging.version import Version
|
|
|
4
4
|
|
|
5
5
|
try: # pragma: no cover
|
|
6
6
|
import cupy
|
|
7
|
+
import cupy.cublas
|
|
7
8
|
import cupyx
|
|
8
9
|
|
|
9
10
|
has_cupy = True
|
|
11
|
+
cublas = cupy.cublas
|
|
10
12
|
cupy_version = Version(cupy.__version__)
|
|
11
13
|
try:
|
|
12
14
|
cupy.cuda.runtime.getDeviceCount()
|
|
@@ -20,6 +22,7 @@ try: # pragma: no cover
|
|
|
20
22
|
else:
|
|
21
23
|
cupy_from_dlpack = cupy.fromDlpack
|
|
22
24
|
except (ImportError, AttributeError):
|
|
25
|
+
cublas = None
|
|
23
26
|
cupy = None
|
|
24
27
|
cupyx = None
|
|
25
28
|
cupy_version = Version("0.0.0")
|
thinc-9.0.0.dev4/thinc/about.py
DELETED
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|