..
experimental
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
torch_cpp_extensions
Run clang-format in CI ( #15524 )
2023-04-18 09:26:58 -07:00
__init__.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_custom_autograd_function.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_custom_autograd_function_exporter.py
Consolidate ORTModule logging ( #16078 )
2023-06-01 10:09:12 +08:00
_custom_autograd_function_runner.py
Adopt linrtunner as the linting tool - take 2 ( #15085 )
2023-03-24 15:29:03 -07:00
_custom_gradient_registry.py
Adopt linrtunner as the linting tool - take 2 ( #15085 )
2023-03-24 15:29:03 -07:00
_custom_op_symbolic_registry.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_execution_agent.py
Adopt linrtunner as the linting tool - take 2 ( #15085 )
2023-03-24 15:29:03 -07:00
_fallback.py
Consolidate ORTModule logging ( #16078 )
2023-06-01 10:09:12 +08:00
_fallback_exceptions.py
Adopt linrtunner as the linting tool - take 2 ( #15085 )
2023-03-24 15:29:03 -07:00
_gradient_accumulation_manager.py
Adopt linrtunner as the linting tool - take 2 ( #15085 )
2023-03-24 15:29:03 -07:00
_graph_execution_interface.py
Adopt linrtunner as the linting tool - take 2 ( #15085 )
2023-03-24 15:29:03 -07:00
_graph_execution_manager.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_graph_execution_manager_factory.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_inference_manager.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_io.py
Consolidate ORTModule logging ( #16078 )
2023-06-01 10:09:12 +08:00
_logger.py
Consolidate ORTModule logging ( #16078 )
2023-06-01 10:09:12 +08:00
_onnx_models.py
Type hint for ORTModule ( #15938 )
2023-05-25 09:28:20 +08:00
_runtime_inspector.py
Introduce memory observer for ORTModule ( #16213 )
2023-06-15 15:45:36 +08:00
_torch_module_factory.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_torch_module_interface.py
Adopt linrtunner as the linting tool - take 2 ( #15085 )
2023-03-24 15:29:03 -07:00
_torch_module_ort.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_torch_module_pytorch.py
Adopt linrtunner as the linting tool - take 2 ( #15085 )
2023-03-24 15:29:03 -07:00
_training_manager.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
_utils.py
Consolidate ORTModule logging ( #16078 )
2023-06-01 10:09:12 +08:00
options.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00
ortmodule.py
Manage ORTModule configurations consistently ( #16396 )
2023-06-27 19:19:36 +08:00