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.CheckpointState method) __getitem__() (onnxruntime.training.api.CheckpointState method) __setitem__() (onnxruntime.training.api.CheckpointState 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) bind_ortvalue_input() (onnxruntime.IOBinding method) bind_ortvalue_output() (onnxruntime.IOBinding method) bind_output() (onnxruntime.IOBinding 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) compile() (onnxruntime.training.ORTModule method) copy_buffer_to_parameters() (onnxruntime.training.api.Module method) copy_outputs_to_cpu() (onnxruntime.IOBinding method) cpu() (onnxruntime.training.ORTModule method) CrossEntropyLoss (onnxruntime.training.artifacts.LossType attribute) cuda() (onnxruntime.training.ORTModule method) custom_metadata_map (onnxruntime.ModelMetadata property) D 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) 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) 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) 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_buffer() (onnxruntime.training.ORTModule method) get_contiguous_parameters() (onnxruntime.training.api.Module method) 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) 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) graph_description (onnxruntime.ModelMetadata property) graph_name (onnxruntime.ModelMetadata property) graph_optimization_level (onnxruntime.SessionOptions property) H half() (onnxruntime.training.ORTModule method) has_value() (onnxruntime.OrtValue method) I 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.NodeArg 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) OrtDevice (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 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) 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.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) SessionOptions (class in onnxruntime) set_extra_state() (onnxruntime.training.ORTModule method) set_learning_rate() (onnxruntime.training.api.Optimizer method) set_providers() (onnxruntime.InferenceSession method) 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) 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)