Jacky Lee 486f023e81 Rename Normalize and move to nn (#513) ...

* Rename Normalize and move to nn

* Match PyTorch for dim>1

2023-02-01 11:55:03 -08:00

..

efficientnet

add ViT test and car

2022-06-05 17:12: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

external_osx_profiling.py

os x profiling: this ratio is exact i believe

2023-01-28 19:02:51 -08:00

external_test_gpu_ast.py

Refactor getenv into helpers (#508)

2023-01-31 15:09:09 -08:00

external_test_image.py

simplify ones after axis splitting

2023-01-14 10:51:43 -08:00

external_test_llvm.py

LLVM Backend take 2 (#403)

2022-10-26 20:32:31 -07:00

external_test_opt.py

one more opt test

2022-10-28 18:37:53 -07:00

graph_batchnorm.py

Simple version of the new GPU backend (#458)

2023-01-10 19:16:02 -08:00

test_conv.py

Fold reduce (#362)

2022-07-19 09:24:02 -07:00

test_efficientnet.py

test_onnx works with enet also

2022-06-11 14:30:26 -07:00

test_gc.py

set requires_grad to None (#387)

2022-09-21 11:16:02 -04:00

test_mnist.py

Refactor getenv into helpers (#508)

2023-01-31 15:09:09 -08:00

test_net_speed.py

import tests from CL metal texture fix

2022-09-19 20:01:47 -04:00

test_nn.py

Rename Normalize and move to nn (#513)

2023-02-01 11:55:03 -08:00

test_onnx.py

support float16 onnx weights (#384)

2022-09-15 09:12:18 -04:00

test_ops.py

A Triton backend for tinygrad (#470)

2023-02-01 11:53:57 -08:00

test_optim.py

tinygrad.nn (#367)

2022-08-18 07:41:00 -07:00

test_shapetracker.py

move shapetracker (#466)

2023-01-19 09:56:31 -08:00

test_speed_v_torch.py

Refactor getenv into helpers (#508)

2023-01-31 15:09:09 -08:00

test_symbolic.py

Add symbolic tests for correctness (#494)

2023-01-30 18:40:16 -08:00

test_tensor.py

set requires_grad to None (#387)

2022-09-21 11:16:02 -04:00

test_train.py

Refactor getenv into helpers (#508)

2023-01-31 15:09:09 -08:00