-
由 Jesse Grabowski 提交于
* Add `OpFromGraph` wrapper around `alloc_diag` * Remove depreciated `AllocDiag` `Op`, rename `AllocDiag2 -> AllocDiag` * Set `inline = False` * Add rewrite to inline all `OpFromGraph` `Op`s * Add `is_zero_offset` helper to `Eye` * Add `is_left_expand_dims` and `is_right_expand_dims` attributes to `DimShuffle` * Seed `test_local_lift_through_linalg` test
| 名称 |
最后提交
|
最后更新 |
|---|---|---|
| .. | ||
| c | ||
| jax | ||
| numba | ||
| pytorch | ||
| __init__.py | ||
| test_link.py | ||
| test_utils.py | ||
| test_vm.py |