Tom EdwardsandGitHub 5bbcbd145c Add cumsum with n-dim inputs (#922) ...

* add cumsum with n-dim inputs, over arbitrary axis + relevant tests

* increased rtol for cumsum test

* move test_cumsum into test_ops

* skip arange test for images as relies on cumsum

* Fix typo

* rewrite cumsum to work with images

2023-06-04 16:55:23 -07:00

..

external

exclude string type from external_test_onnx_backend.py (#918)

2023-06-03 19:10:52 -07:00

extra

Refactor LoadOps (#910)

2023-06-03 09:40:43 -07:00

models

Zero dim Tensor support (#777)

2023-06-01 11:32:02 -07:00

unit

Refactor LoadOps (#910)

2023-06-03 09:40:43 -07:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

Dockerfile

testdocker

2023-01-09 12:41:52 -08:00

test_assign.py

Refactor LoadOps (#910)

2023-06-03 09:40:43 -07:00

test_conv_shapetracker.py

speed like you wouldn't believe (#626)

2023-03-02 07:49:19 -08:00

test_conv.py

Add conv binops_no_rerun test assertions (#665)

2023-03-10 19:09:48 -08:00

test_custom_function.py

Disktensors! (#819)

2023-05-28 15:40:37 -07:00

test_dtype.py

Llvm int support (#866)

2023-05-30 17:49:26 -07:00

test_gc.py

fix up eye and fix gc test

2023-02-27 06:53:18 -08:00

test_jit.py

assertions for jit

2023-05-05 21:56:32 -07:00

test_net_speed.py

test touchups

2023-05-26 02:39:42 +00:00

test_nn.py

support Int64 type in cstyle gen (#860)

2023-05-30 16:04:46 -07:00

test_ops.py

Add cumsum with n-dim inputs (#922)

2023-06-04 16:55:23 -07:00

test_optim.py

Add weight decay to SGD (#883)

2023-06-01 13:13:18 -07:00

test_randomness.py

Refactor LoadOps (#910)

2023-06-03 09:40:43 -07:00

test_specific_conv.py

disable that test on LLVM. i have to stop pushing to master

2023-05-26 03:11:03 +00:00

test_speed_v_torch.py

good changes from the M1 Tensor Core project (#730)

2023-03-29 05:11:02 +04:00

test_tensor.py

Zero dim Tensor support (#777)

2023-06-01 11:32:02 -07:00