| .. |
|
experimental
|
Disable PERF* rules in ruff to allow better readability (#16834)
|
2023-07-25 15:38:22 -07:00 |
|
torch_cpp_extensions
|
Disable PERF* rules in ruff to allow better readability (#16834)
|
2023-07-25 15:38:22 -07:00 |
|
__init__.py
|
Add mac and windows python packages for onnxruntime-training (#16993)
|
2023-08-07 20:32:55 -07:00 |
|
_custom_autograd_function.py
|
Manage ORTModule configurations consistently (#16396)
|
2023-06-27 19:19:36 +08:00 |
|
_custom_autograd_function_exporter.py
|
Fix few small bugs (#17019)
|
2023-08-07 14:01:36 +08:00 |
|
_custom_autograd_function_runner.py
|
|
|
|
_custom_gradient_registry.py
|
[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)
|
2023-07-21 12:53:41 -07:00 |
|
_custom_op_symbolic_registry.py
|
[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)
|
2023-07-21 12:53:41 -07:00 |
|
_execution_agent.py
|
|
|
|
_fallback.py
|
Consolidate ORTModule logging (#16078)
|
2023-06-01 10:09:12 +08:00 |
|
_fallback_exceptions.py
|
|
|
|
_gradient_accumulation_manager.py
|
|
|
|
_graph_execution_interface.py
|
|
|
|
_graph_execution_manager.py
|
Refactor schema extraction and output unflattening (#16894)
|
2023-08-04 13:58:21 +08:00 |
|
_graph_execution_manager_factory.py
|
ORTModule log clean up (#16795)
|
2023-07-26 12:42:50 +08:00 |
|
_inference_manager.py
|
Refactor schema extraction and output unflattening (#16894)
|
2023-08-04 13:58:21 +08:00 |
|
_io.py
|
Refactor schema extraction and output unflattening (#16894)
|
2023-08-04 13:58:21 +08:00 |
|
_logger.py
|
Fix few small bugs (#17019)
|
2023-08-07 14:01:36 +08:00 |
|
_onnx_models.py
|
Save optimized pre_grad graph once ready (#16816)
|
2023-08-02 14:05:26 +08:00 |
|
_runtime_inspector.py
|
ORTModule log clean up (#16795)
|
2023-07-26 12:42:50 +08:00 |
|
_torch_module_factory.py
|
Manage ORTModule configurations consistently (#16396)
|
2023-06-27 19:19:36 +08:00 |
|
_torch_module_interface.py
|
|
|
|
_torch_module_ort.py
|
ORTModule log clean up (#16795)
|
2023-07-26 12:42:50 +08:00 |
|
_torch_module_pytorch.py
|
|
|
|
_training_manager.py
|
Refactor schema extraction and output unflattening (#16894)
|
2023-08-04 13:58:21 +08:00 |
|
_utils.py
|
Refactor schema extraction and output unflattening (#16894)
|
2023-08-04 13:58:21 +08:00 |
|
graph_transformer_registry.py
|
[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)
|
2023-07-21 12:53:41 -07:00 |
|
options.py
|
Fix few small bugs (#17019)
|
2023-08-07 14:01:36 +08:00 |
|
ortmodule.py
|
ORTModule log clean up (#16795)
|
2023-07-26 12:42:50 +08:00 |
