xla
[FSDP] Add profiler to ImageNet
#4754
Merged

Commits
  • [FSDP] Add profiler to ImageNet
    alanwaketan committed 3 years ago
  • Fix linters
    alanwaketan committed 3 years ago
  • Move add_step_closure out from build_graph
    alanwaketan committed 3 years ago
Loading