tinygrad

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-23 13:58:00 -05:00

Files

Joqsan ef129bcb85 Zero dim Tensor support (#777 )

* add and reorganize test_slice_* tests

* refactor Tensor.__getitem__()

* preliminary tests for 1) 0D tensors and 2) varargs for Tensor.zeros and Tensor.ones

* always compare shapes of the numpy arrays obtained from tinygrad and torch tensors

* add more tests for 0D support

* remove test_tensor.test_slicing(). All slicing tests at test/test_ops.py

* add zero-dim support

* make test_end2end.py consistent with 0dim support

* add test for tensor with zero in shape

* don't simplify ones if shape is ()

* skip tests that need zero-size tensor support.

- zero-size tensor support not related to 0dim tensors.

* add tests for __getitem__() supporting strides >= 1

* refactor __getitem__: support for strides >= 1

* minor refactors and add comments to __getitem__

* add tests for slices with negative steps

* add support for slices with negative strides

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

efficientnet

test cleanups

2023-03-03 06:36:06 -08:00

test_efficientnet.py

Inference test: add tests for ResNet50 (#773 )

2023-05-13 21:18:15 -07:00

test_end2end.py

Zero dim Tensor support (#777 )

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

test_mnist.py

remove RMSprop, nobody uses it anymore