| .. |
|
external
|
Add sin/cos/tan (#794)
|
2023-05-25 09:04:56 -07:00 |
|
extra
|
linearizer! (#714)
|
2023-03-19 23:43:49 -07:00 |
|
models
|
Add mlperf RNN-T model (#782)
|
2023-05-25 00:41:21 -07:00 |
|
unit
|
Fix PytestCollectionWarning when running tests (#791)
|
2023-05-24 23:17:57 -07:00 |
|
Dockerfile
|
testdocker
|
2023-01-09 12:41:52 -08:00 |
|
__init__.py
|
All devices are equal! (#196)
|
2020-12-15 23:44:08 -08:00 |
|
test_assign.py
|
assign buffer reuse (#547)
|
2023-02-09 11:53:02 -06:00 |
|
test_conv.py
|
Add conv binops_no_rerun test assertions (#665)
|
2023-03-10 19:09:48 -08:00 |
|
test_conv_shapetracker.py
|
speed like you wouldn't believe (#626)
|
2023-03-02 07:49:19 -08:00 |
|
test_custom_function.py
|
fix binop, other tests failure (#723)
|
2023-03-22 18:15:07 -07:00 |
|
test_dtype.py
|
casting: support all backends and implement half (#726)
|
2023-03-24 09:58:03 -07:00 |
|
test_gc.py
|
fix up eye and fix gc test
|
2023-02-27 06:53:18 -08:00 |
|
test_jit.py
|
assertions for jit
|
2023-05-05 21:56:32 -07:00 |
|
test_net_speed.py
|
rename log_softmax, support dim, fix onnx Softmax
|
2023-02-24 10:11:24 -08:00 |
|
test_nn.py
|
LayerNorm2d for 2 lines
|
2023-03-20 16:58:43 -07:00 |
|
test_ops.py
|
Add sin/cos/tan (#794)
|
2023-05-25 09:04:56 -07:00 |
|
test_optim.py
|
remove RMSprop, nobody uses it anymore
|
2023-03-20 12:31:34 -07:00 |
|
test_randomness.py
|
Statistics test: check if distributions match torch (#769)
|
2023-05-07 21:43:23 -07:00 |
|
test_specific_conv.py
|
clean up opt (#649)
|
2023-03-05 20:49:36 -08:00 |
|
test_speed_v_torch.py
|
good changes from the M1 Tensor Core project (#730)
|
2023-03-29 05:11:02 +04:00 |
|
test_tensor.py
|
ConvNeXt + extras (#657)
|
2023-03-06 22:10:56 -08:00 |
c19ef0fcce