Index _ | A | B | C | D | E | F | G | H | I | L | M | N | O | P | R | S | T | U | V | X | Z _ __call__() (onnxruntime.training.api.Module method) __contains__() (onnxruntime.training.api.checkpoint_state.Parameters method) (onnxruntime.training.api.checkpoint_state.Properties method) __getitem__() (onnxruntime.training.api.checkpoint_state.Parameters method) (onnxruntime.training.api.checkpoint_state.Properties method) __iter__() (onnxruntime.training.api.checkpoint_state.Parameters method) (onnxruntime.training.api.checkpoint_state.Properties method) __len__() (onnxruntime.training.api.checkpoint_state.Parameters method) (onnxruntime.training.api.checkpoint_state.Properties method) __repr__() (onnxruntime.training.api.checkpoint_state.Parameter method) (onnxruntime.training.api.checkpoint_state.Parameters method) (onnxruntime.training.api.checkpoint_state.Properties method) __setitem__() (onnxruntime.training.api.checkpoint_state.Parameters method) (onnxruntime.training.api.checkpoint_state.Properties method) A AdamW (onnxruntime.training.artifacts.OptimType attribute) add_external_initializers() (onnxruntime.SessionOptions method) add_free_dimension_override_by_denotation() (onnxruntime.SessionOptions method) add_free_dimension_override_by_name() (onnxruntime.SessionOptions method) add_initializer() (onnxruntime.SessionOptions method) add_module() (onnxruntime.training.ORTModule method) add_run_config_entry() (onnxruntime.RunOptions method) add_session_config_entry() (onnxruntime.SessionOptions method) apply() (onnxruntime.training.ORTModule method) as_blocksparse_view() (onnxruntime.SparseTensor method) as_coo_view() (onnxruntime.SparseTensor method) as_csrc_view() (onnxruntime.SparseTensor method) as_sparse_tensor() (onnxruntime.OrtValue method) B base (onnxruntime.training.onnxblock.Block attribute) BCEWithLogitsLoss (onnxruntime.training.artifacts.LossType attribute) bfloat16() (onnxruntime.training.ORTModule method) bind_cpu_input() (onnxruntime.IOBinding method) bind_input() (onnxruntime.IOBinding method) (onnxruntime.SessionIOBinding method) bind_ortvalue_input() (onnxruntime.IOBinding method) (onnxruntime.SessionIOBinding method) bind_ortvalue_output() (onnxruntime.IOBinding method) (onnxruntime.SessionIOBinding method) bind_output() (onnxruntime.IOBinding method) (onnxruntime.SessionIOBinding method) Block (class in onnxruntime.training.onnxblock) buffers() (onnxruntime.training.ORTModule method) build() (onnxruntime.training.onnxblock.Block method) (onnxruntime.training.onnxblock.ForwardBlock method) (onnxruntime.training.onnxblock.TrainingBlock method) C CheckpointState (class in onnxruntime.training.api) children() (onnxruntime.training.ORTModule method) clear_binding_inputs() (onnxruntime.SessionIOBinding method) clear_binding_outputs() (onnxruntime.SessionIOBinding method) compile() (onnxruntime.training.ORTModule method) copy_buffer_to_parameters() (onnxruntime.training.api.Module method) copy_outputs_to_cpu() (onnxruntime.IOBinding method) (onnxruntime.SessionIOBinding method) cpu() (onnxruntime.training.ORTModule method) create_and_register_allocator() (in module onnxruntime) create_and_register_allocator_v2() (in module onnxruntime) CrossEntropyLoss (onnxruntime.training.artifacts.LossType attribute) cuda() (onnxruntime.training.ORTModule method) custom_metadata_map (onnxruntime.ModelMetadata property) D data (onnxruntime.training.api.checkpoint_state.Parameter property) data_ptr() (onnxruntime.OrtValue method) data_type() (onnxruntime.OrtValue method) (onnxruntime.SparseTensor method) dense_shape() (onnxruntime.SparseTensor method) description (onnxruntime.ModelMetadata property) device_name() (onnxruntime.OrtValue method) (onnxruntime.SparseTensor method) disable_fallback() (onnxruntime.InferenceSession method) disable_telemetry_events() (in module onnxruntime) domain (onnxruntime.ModelMetadata property) double() (onnxruntime.training.ORTModule method) E element_type() (onnxruntime.OrtValue method) enable_cpu_mem_arena (onnxruntime.SessionOptions property) enable_fallback() (onnxruntime.InferenceSession method) enable_mem_pattern (onnxruntime.SessionOptions property) enable_mem_reuse (onnxruntime.SessionOptions property) enable_profiling (onnxruntime.SessionOptions property) enable_telemetry_events() (in module onnxruntime) end_profiling() (onnxruntime.InferenceSession method) eval() (onnxruntime.training.api.Module method) (onnxruntime.training.ORTModule method) execution_mode (onnxruntime.SessionOptions property) execution_order (onnxruntime.SessionOptions property) ExecutionMode (class in onnxruntime) ExecutionOrder (class in onnxruntime) export_model_for_inferencing() (onnxruntime.training.api.Module method) extra_repr() (onnxruntime.training.ORTModule method) F float() (onnxruntime.training.ORTModule method) format() (onnxruntime.SparseTensor method) forward() (onnxruntime.training.ORTModule method) ForwardBlock (class in onnxruntime.training.onnxblock) G generate_artifacts() (in module onnxruntime.training.artifacts) get_all_providers() (in module onnxruntime) get_available_providers() (in module onnxruntime) get_buffer() (onnxruntime.training.ORTModule method) get_build_info() (in module onnxruntime) get_contiguous_parameters() (onnxruntime.training.api.Module method) get_device() (in module onnxruntime) get_extra_state() (onnxruntime.training.ORTModule method) get_inputs() (onnxruntime.InferenceSession method) get_learning_rate() (onnxruntime.training.api.Optimizer method) get_modelmeta() (onnxruntime.InferenceSession method) get_outputs() (onnxruntime.InferenceSession method) (onnxruntime.IOBinding method) (onnxruntime.SessionIOBinding method) get_overridable_initializers() (onnxruntime.InferenceSession method) get_parameter() (onnxruntime.training.ORTModule method) get_parameters_size() (onnxruntime.training.api.Module method) get_profiling_start_time_ns() (onnxruntime.InferenceSession method) get_provider_options() (onnxruntime.InferenceSession method) get_providers() (onnxruntime.InferenceSession method) get_run_config_entry() (onnxruntime.RunOptions method) get_session_config_entry() (onnxruntime.SessionOptions method) get_session_options() (onnxruntime.InferenceSession method) get_submodule() (onnxruntime.training.ORTModule method) get_version_string() (in module onnxruntime) grad (onnxruntime.training.api.checkpoint_state.Parameter property) graph_description (onnxruntime.ModelMetadata property) graph_name (onnxruntime.ModelMetadata property) graph_optimization_level (onnxruntime.SessionOptions property) GraphOptimizationLevel (class in onnxruntime) H half() (onnxruntime.training.ORTModule method) has_collective_ops() (in module onnxruntime) has_value() (onnxruntime.OrtValue method) I infer_shapes_on_base() (onnxruntime.training.onnxblock.Block method) (onnxruntime.training.onnxblock.ForwardBlock method) (onnxruntime.training.onnxblock.TrainingBlock method) InferenceSession (class in onnxruntime) input_names() (onnxruntime.training.api.Module method) inter_op_num_threads (onnxruntime.SessionOptions property) intra_op_num_threads (onnxruntime.SessionOptions property) io_binding() (onnxruntime.InferenceSession method) IOBinding (class in onnxruntime) ipu() (onnxruntime.training.ORTModule method) is_compatible() (in module onnxruntime.backend) is_sparse_tensor() (onnxruntime.OrtValue method) is_tensor() (onnxruntime.OrtValue method) is_tensor_sequence() (onnxruntime.OrtValue method) L L1Loss (onnxruntime.training.artifacts.LossType attribute) lazy_reset_grad() (onnxruntime.training.api.Module method) LinearLRScheduler (class in onnxruntime.training.api) load_checkpoint() (onnxruntime.training.api.CheckpointState class method) load_state_dict() (onnxruntime.training.ORTModule method) log_severity_level (onnxruntime.RunOptions property) (onnxruntime.SessionOptions property) log_verbosity_level (onnxruntime.RunOptions property) (onnxruntime.SessionOptions property) logid (onnxruntime.RunOptions property) (onnxruntime.SessionOptions property) LossType (class in onnxruntime.training.artifacts) M ModelMetadata (class in onnxruntime) Module (class in onnxruntime.training.api) module (onnxruntime.training.ORTModule property) modules() (onnxruntime.training.ORTModule method) MSELoss (onnxruntime.training.artifacts.LossType attribute) N name (onnxruntime.ExecutionMode property) (onnxruntime.ExecutionOrder property) (onnxruntime.GraphOptimizationLevel property) (onnxruntime.NodeArg property) (onnxruntime.OrtAllocatorType property) (onnxruntime.OrtMemType property) (onnxruntime.training.api.checkpoint_state.Parameter property) named_buffers() (onnxruntime.training.ORTModule method) named_children() (onnxruntime.training.ORTModule method) named_modules() (onnxruntime.training.ORTModule method) named_parameters() (onnxruntime.training.ORTModule method) NodeArg (class in onnxruntime) numpy() (onnxruntime.OrtValue method) O only_execute_path_to_fetches (onnxruntime.RunOptions property) optimized_model_filepath (onnxruntime.SessionOptions property) Optimizer (class in onnxruntime.training.api) OptimType (class in onnxruntime.training.artifacts) ort_value_from_sparse_tensor() (onnxruntime.OrtValue static method) OrtAllocatorType (class in onnxruntime) OrtArenaCfg (class in onnxruntime) OrtDevice (class in onnxruntime) OrtMemoryInfo (class in onnxruntime) OrtMemType (class in onnxruntime) ORTModule (class in onnxruntime.training) OrtValue (class in onnxruntime) ortvalue_from_numpy() (onnxruntime.OrtValue static method) ortvalue_from_shape_and_type() (onnxruntime.OrtValue static method) output_names() (onnxruntime.training.api.Module method) P Parameter (class in onnxruntime.training.api.checkpoint_state) Parameters (class in onnxruntime.training.api.checkpoint_state) parameters (onnxruntime.training.api.CheckpointState property) parameters() (onnxruntime.training.onnxblock.TrainingBlock method) (onnxruntime.training.ORTModule method) prepare() (in module onnxruntime.backend) producer_name (onnxruntime.ModelMetadata property) profile_file_prefix (onnxruntime.SessionOptions property) Properties (class in onnxruntime.training.api.checkpoint_state) properties (onnxruntime.training.api.CheckpointState property) R register_backward_hook() (onnxruntime.training.ORTModule method) register_buffer() (onnxruntime.training.ORTModule method) register_custom_ops_library() (onnxruntime.SessionOptions method) register_forward_hook() (onnxruntime.training.ORTModule method) register_forward_pre_hook() (onnxruntime.training.ORTModule method) register_full_backward_hook() (onnxruntime.training.ORTModule method) register_full_backward_pre_hook() (onnxruntime.training.ORTModule method) register_load_state_dict_post_hook() (onnxruntime.training.ORTModule method) register_module() (onnxruntime.training.ORTModule method) register_parameter() (onnxruntime.training.ORTModule method) register_state_dict_pre_hook() (onnxruntime.training.ORTModule method) requires_grad (onnxruntime.training.api.checkpoint_state.Parameter property) requires_grad() (onnxruntime.training.onnxblock.TrainingBlock method) requires_grad_() (onnxruntime.training.ORTModule method) run() (in module onnxruntime.backend) (onnxruntime.InferenceSession method) run_async() (onnxruntime.InferenceSession method) run_with_iobinding() (onnxruntime.InferenceSession method) run_with_ort_values() (onnxruntime.InferenceSession method) run_with_ortvaluevector() (onnxruntime.InferenceSession method) RunOptions (class in onnxruntime) S save_checkpoint() (onnxruntime.training.api.CheckpointState class method) SessionIOBinding (class in onnxruntime) SessionOptions (class in onnxruntime) set_default_logger_severity() (in module onnxruntime) set_default_logger_verbosity() (in module onnxruntime) set_extra_state() (onnxruntime.training.ORTModule method) set_learning_rate() (onnxruntime.training.api.Optimizer method) set_providers() (onnxruntime.InferenceSession method) set_seed() (in module onnxruntime) SGD (onnxruntime.training.artifacts.OptimType attribute) shape (onnxruntime.NodeArg property) shape() (onnxruntime.OrtValue method) share_memory() (onnxruntime.training.ORTModule method) sparse_coo_from_numpy() (onnxruntime.SparseTensor static method) sparse_csr_from_numpy() (onnxruntime.SparseTensor static method) SparseTensor (class in onnxruntime) state_dict() (onnxruntime.training.ORTModule method) step() (onnxruntime.training.api.LinearLRScheduler method) (onnxruntime.training.api.Optimizer method) supports_device() (in module onnxruntime.backend) synchronize_inputs() (onnxruntime.SessionIOBinding method) synchronize_outputs() (onnxruntime.SessionIOBinding method) T terminate (onnxruntime.RunOptions property) to() (onnxruntime.training.ORTModule method) to_cuda() (onnxruntime.SparseTensor method) to_empty() (onnxruntime.training.ORTModule method) to_model_proto() (onnxruntime.training.onnxblock.ForwardBlock method) (onnxruntime.training.onnxblock.TrainingBlock method) train() (onnxruntime.training.api.Module method) (onnxruntime.training.ORTModule method) training (onnxruntime.training.api.Module attribute) training_mode (onnxruntime.RunOptions property) TrainingBlock (class in onnxruntime.training.onnxblock) type (onnxruntime.NodeArg property) type() (onnxruntime.training.ORTModule method) U update_inplace() (onnxruntime.OrtValue method) use_deterministic_compute (onnxruntime.SessionOptions property) V values() (onnxruntime.SparseTensor method) version (onnxruntime.ModelMetadata property) X xpu() (onnxruntime.training.ORTModule method) Z zero_grad() (onnxruntime.training.ORTModule method)