tinygrad/test/models
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 2023-03-20 12:31:34 -07:00
test_onnx.py fix external test + speed 2023-03-03 06:46:16 -08:00
test_rnnt.py make tests faster + add onnx (#815) 2023-05-27 08:53:32 -07:00
test_train.py ConvNeXt + extras (#657) 2023-03-06 22:10:56 -08:00