tinygrad/tinygrad
Umut Zengin 35bf21276f
Argmax/Argmin Feature (#1576)
* implemented argmax and argmin

* lint

* lint

* match torch behaviour

* format

* removed flip
2023-08-20 18:46:46 -07:00
..
codegen not worth the speed (#1584) 2023-08-20 10:24:58 -07:00
nn Add assert for wrong matmul/dot shapes (#1438) 2023-08-04 18:16:56 -04:00
renderer fix __launch_bounds__ and benchmark TC MATMUL (#1575) 2023-08-19 10:54:39 -07:00
runtime fix __launch_bounds__ and benchmark TC MATMUL (#1575) 2023-08-19 10:54:39 -07:00
shape more symbolic symbolic ops (#1564) 2023-08-18 09:21:41 -07:00
graph.py Add WHERE ternary (or trinary?) op (#1196) 2023-07-16 00:31:55 -07:00
helpers.py teeny changes (#1589) 2023-08-20 13:38:38 -07:00
jit.py Symbolic Shape JIT main PR (#1353) 2023-08-18 14:39:55 -07:00
lazy.py teeny changes (#1589) 2023-08-20 13:38:38 -07:00
mlops.py teeny changes (#1589) 2023-08-20 13:38:38 -07:00
ops.py not worth the speed (#1584) 2023-08-20 10:24:58 -07:00
state.py add stable diffusion and llama (#1471) 2023-08-06 21:31:51 -07:00
tensor.py Argmax/Argmin Feature (#1576) 2023-08-20 18:46:46 -07:00