Ubaidullah Khan c825cc4774 use tensor dtype for zeros_like() (#842) ...

* use tensor dtype for zeros_like()

* add tests for zeros_like dtype

* iterate over dtypes

* remove space

* remove print

* fix test, iterate over a list

2023-05-29 08:05:50 -07:00

..

external

Add LAMB optimizer (#821)

2023-05-28 15:09:05 -07:00

extra

make tests faster + add onnx (#815)

2023-05-27 08:53:32 -07:00

models

make tests faster + add onnx (#815)

2023-05-27 08:53:32 -07:00

unit

Disktensors! (#819)

2023-05-28 15:40:37 -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

assign buffer reuse (#547)

2023-02-09 11:53:02 -06: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

int8/uint8 support (#837)

2023-05-28 23:15:06 -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

Add MLPerf UNet3D model (#775)

2023-05-28 20:38:19 -07:00

test_ops.py

Fix conv_transpose2d asymmetric padding (#840)

2023-05-29 07:57:06 -07:00

test_optim.py

remove RMSprop, nobody uses it anymore

2023-03-20 12:31:34 -07:00

test_randomness.py

Statistics test: check if distributions match torch (#769)

2023-05-07 21:43:23 -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

use tensor dtype for zeros_like() (#842)

2023-05-29 08:05:50 -07:00