Jacky LeeandGitHub d768acb6a9 Rename Normalize and move to nn (#415) ...

* Rename Normalize and move to nn

* Fix comparison to None error

* Add test for GroupNorm

* Rename test case

* Flip parameters to match PyTorch

* Increase error tolerance

* Fix elementwise_affine on channels

* Match arguments with PyTorch

* Initialize weight and bias only when affine is true

* Is this it?

* A bit cleaner

* Handle case where weight or bias is None

2023-01-25 07:47:59 -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_test_gpu_ast.py

the ast test is actually a test now

2023-01-24 07:53:24 -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

lib_test_ast.py

the ast test is actually a test now

2023-01-24 07:53:24 -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

Fix weight init: this work? (#391)

2022-09-25 16:46:33 -04: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 (#415)

2023-01-25 07:47:59 -08:00

test_onnx.py

support float16 onnx weights (#384)

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

test_ops.py

indexer changes for master

2023-01-18 18:02:02 -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

delete opencl <celebration>

2023-01-24 14:18:35 -08:00

test_symbolic.py

Typing (#468)

2023-01-21 09:09:22 -08:00

test_tensor.py

set requires_grad to None (#387)

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

test_train.py

Simple version of the new GPU backend (#458)

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