flax
Add clarifying docstring for 'size' argument to prefetch_to_device's
#1574
Merged

Commits
  • Add clarifying docstring for 'size' argument to prefetch_to_device's
    avital committed 4 years ago
  • Merge branch 'main' into avital-prefetch-doc
    marcvanzee committed 4 years ago
Loading