| .. |
|
experimental
|
|
|
|
graph_optimizers
|
|
|
|
torch_cpp_extensions
|
|
|
|
__init__.py
|
|
|
|
_custom_autograd_function.py
|
|
|
|
_custom_autograd_function_exporter.py
|
|
|
|
_custom_autograd_function_runner.py
|
|
|
|
_custom_gradient_registry.py
|
|
|
|
_custom_op_symbolic_registry.py
|
|
|
|
_execution_agent.py
|
|
|
|
_fallback.py
|
|
|
|
_fallback_exceptions.py
|
|
|
|
_gradient_accumulation_manager.py
|
|
|
|
_graph_execution_interface.py
|
|
|
|
_graph_execution_manager.py
|
|
|
|
_graph_execution_manager_factory.py
|
|
|
|
_inference_manager.py
|
|
|
|
_io.py
|
|
|
|
_logger.py
|
|
|
|
_onnx_models.py
|
|
|
|
_runtime_inspector.py
|
|
|
|
_torch_module_factory.py
|
|
|
|
_torch_module_interface.py
|
|
|
|
_torch_module_ort.py
|
|
|
|
_torch_module_pytorch.py
|
|
|
|
_training_manager.py
|
|
|
|
_utils.py
|
|
|
|
_zero_stage3_compatibility.py
|
|
|
|
graph_optimizer_registry.py
|
|
|
|
options.py
|
|
|
|
ortmodule.py
|
|
|