onnxruntime
2347a0ac - Autograd Function Fallback bug fix - moe support (#8105)

Commit
4 years ago
Autograd Function Fallback bug fix - moe support (#8105) * Support forward inputs orders like "Non_tensor/Tensor/Non_tensor". Correspondingly, support "None/Tensor_Grad/None" fpr backward outputs. * Report RuntimeError when PythonOp detected but _enable_custom_autograd_function is enabled. * Fix "PoliCheck ] - Defect : Term "hang", Component : orttraining\orttraining\python\training\ortmodule\__init__.py (1 issue)" * rename call_convention->input_convention, input_tensor_requires_grads->input_requires_grads * fix minor comment * revert polycheck fix in case of conflict * Update orttraining/orttraining/core/graph/training_op_defs.cc Co-authored-by: Tim Harris <tiharr@microsoft.com> * Apply suggestions from code review Refine the schema description Co-authored-by: Tim Harris <tiharr@microsoft.com> * Resolve review comments Co-authored-by: Tim Harris <tiharr@microsoft.com>
Author
Parents
Loading