[Pytorch] Enable TE Sequential Op to consume extra_outputs from a previously run Op - #3320

[Pytorch] Enable TE Sequential Op to consume extra_outputs from a previously run Op#3320

vthumbe1503 wants to merge 13 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

vthumbe1503:enable_extra_out_consumptionvthumbe1503/TransformerEngine:enable_extra_out_consumptionCopy head branch name to clipboard