Mitchell Goff ae4f0aeb5f NumPy-like semantics for Tensor.__getitem__ (#506) ...

* Rewrote Tensor.__getitem__ to fix negative indices and add support for np.newaxis/None

* Fixed pad2d

* mypy doesn't know about mlops methods

* normal python behavior for out-of-bounds slicing

* type: ignore

* inlined idxfix

* added comment for __getitem__

* Better comments, better tests, and fixed bug in np.newaxis

2023-02-08 08:59:46 -06: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

Add test for quick_gelu (#526)

2023-02-03 20:01:39 -08:00

test_optim.py

tinygrad.nn (#367)

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

test_shapetracker.py

remove val expansion (#539)

2023-02-07 15:14:05 -06:00

test_speed_v_torch.py

Refactor getenv into helpers (#508)

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

test_symbolic.py

more symbolic correctness

2023-02-07 00:03:14 -06:00

test_tensor.py

NumPy-like semantics for Tensor.__getitem__ (#506)

2023-02-08 08:59:46 -06:00

test_train.py

test_train: Adam -> SGD

2023-02-06 08:55:41 -06:00