Philippe Tillet b0647cfd52 [PYTHON] Removed support for dense softmax ...

Interest seems limited now that it is fused in cross_entropy. Will
likely re-add once it's easier to share code between ops

2021-07-27 12:38:48 -07:00

..

test_blocksparse.py

[PYTHON][OPS][BLOCKSPARSE] Now rounding softmax tile sizes to next power

2021-07-27 12:38:48 -07:00

test_conv.py

[PYTHON][TESTS][DOC] Various improvement of the API and code quality:

2021-07-27 12:38:48 -07:00

test_cross_entropy.py

Added a Softmax Xent Op (#53)

2021-07-27 12:38:48 -07:00

test_matmul.py

[TESTS] test_matmul.py now plots benchmarks

2021-07-27 12:38:48 -07:00