search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k stars 483 forks source link

Migrate from capture_pre_autograd_graph to export_for_training #8398

Open yushangdi opened 6 days ago

yushangdi commented 6 days ago

capture_pre_autograd_graph API is deprecated and will be deleted.

yushangdi commented 6 days ago

@lsy323 @qihqi Hi, the capture_pre_autograd_graph API is deprecated and will be deleted soon. Please use export_for_training instead. Thanks!