transformers
105c3a48 - Support extraction of both train and eval XLA graphs (#20492)

Commit
3 years ago
Support extraction of both train and eval XLA graphs (#20492) Neuron supports extraction of XLA graphs for compilation. However, when both do_train and do_eval options are enabled, sizes returned by tensor operator can be 0. To avoid INVALID_ARGUMENT error, we use inequality in the check whether a tensor needs padding or not.
Author
Parents
Loading