xla
0cd1fc2a - Fix dk/dv autograd error on TPU flash attention (#8685)

Commit

1 year ago

Fix dk/dv autograd error on TPU flash attention (#8685)

References

#8685 - Fix dk/dv autograd error on TPU flash attention

Author

zmelumian972

zmelumian972

Parents

Loading