ORTModule
1.14.92+cpu
Contents:
Overview
Gallery of examples
API
ORTModule
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Z
A
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)
allow (onnxruntime.capi._pybind_state.PropagateCastOpsConfiguration property)
allreduce_post_accumulation (onnxruntime.TrainingParameters property)
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)
attn_dropout_recompute (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property)
(onnxruntime.TrainingParameters property)
attributes (onnxruntime.capi._pybind_state.GradientNodeDefinition property)
B
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)
bool_tensor_indices() (onnxruntime.capi._pybind_state.OrtValueVector method)
buffers() (onnxruntime.training.ORTModule method)
build() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method)
build_gradient_graph (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property)
C
cached_node_arg_names (onnxruntime.capi._pybind_state.GraphInfo property)
children() (onnxruntime.training.ORTModule method)
clear() (onnxruntime.capi._pybind_state.OrtValueCache method)
copy_outputs_to_cpu() (onnxruntime.IOBinding method)
count() (onnxruntime.capi._pybind_state.OrtValueCache method)
cpu() (onnxruntime.training.ORTModule method)
cuda() (onnxruntime.training.ORTModule method)
D
data_parallel_size (onnxruntime.TrainingParameters property)
data_ptr() (onnxruntime.OrtValue method)
data_type() (onnxruntime.OrtValue method)
(onnxruntime.SparseTensor method)
deepspeed_zero_stage (onnxruntime.TrainingParameters property)
dense_shape() (onnxruntime.SparseTensor method)
device_name() (onnxruntime.OrtValue method)
(onnxruntime.SparseTensor method)
disable_fallback() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
dlpack_at() (onnxruntime.capi._pybind_state.OrtValueVector method)
domain (onnxruntime.capi._pybind_state.GradientNodeDefinition property)
double() (onnxruntime.training.ORTModule method)
dtype (onnxruntime.capi._pybind_state.GradientNodeAttributeDefinition property)
dump_patches (onnxruntime.training.ORTModule attribute)
E
element_type() (onnxruntime.OrtValue method)
element_type_at() (onnxruntime.capi._pybind_state.OrtValueVector method)
enable_adasum (onnxruntime.TrainingParameters property)
enable_caching (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property)
enable_cpu_mem_arena (onnxruntime.SessionOptions property)
enable_fallback() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
enable_gelu_approximation (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property)
enable_grad_norm_clip (onnxruntime.TrainingParameters property)
enable_mem_pattern (onnxruntime.SessionOptions property)
enable_mem_reuse (onnxruntime.SessionOptions property)
enable_profiling (onnxruntime.SessionOptions property)
end_profiling() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
eval() (onnxruntime.training.ORTModule method)
eval_step() (onnxruntime.capi.ort_trainer.ORTTrainer method)
(onnxruntime.training.ORTTrainer method)
execution_mode (onnxruntime.SessionOptions property)
execution_order (onnxruntime.SessionOptions property)
extra_repr() (onnxruntime.training.ORTModule method)
F
float() (onnxruntime.training.ORTModule method)
format() (onnxruntime.SparseTensor method)
forward() (onnxruntime.training.ORTModule method)
frontier_node_arg_map (onnxruntime.capi._pybind_state.GraphInfo property)
G
gelu_recompute (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property)
(onnxruntime.TrainingParameters property)
get_buffer() (onnxruntime.training.ORTModule method)
get_extra_state() (onnxruntime.training.ORTModule method)
get_forward_model() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method)
get_gradient_model() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method)
get_graph_info() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method)
get_inputs() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
get_modelmeta() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
get_outputs() (onnxruntime.InferenceSession method)
(onnxruntime.IOBinding method)
(onnxruntime.TrainingSession method)
get_overridable_initializers() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
get_parameter() (onnxruntime.training.ORTModule method)
get_profiling_start_time_ns() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
get_provider_options() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
get_providers() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
get_run_config_entry() (onnxruntime.RunOptions method)
get_session_config_entry() (onnxruntime.SessionOptions method)
get_session_options() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
get_submodule() (onnxruntime.training.ORTModule method)
gradient_accumulation_steps (onnxruntime.TrainingParameters property)
GradientNodeAttributeDefinition (class in onnxruntime.capi._pybind_state)
GradientNodeDefinition (class in onnxruntime.capi._pybind_state)
graph_optimization_level (onnxruntime.SessionOptions property)
graph_transformer_config (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property)
GraphInfo (class in onnxruntime.capi._pybind_state)
GraphTransformerConfiguration (class in onnxruntime.capi._pybind_state)
H
half() (onnxruntime.training.ORTModule method)
has_value() (onnxruntime.OrtValue method)
horizontal_parallel_size (onnxruntime.TrainingParameters property)
I
immutable_weights (onnxruntime.TrainingParameters property)
InferenceSession (class in onnxruntime)
initialize() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method)
initializer_grad_names_to_train (onnxruntime.capi._pybind_state.GraphInfo property)
initializer_names (onnxruntime.capi._pybind_state.GraphInfo property)
(onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property)
initializer_names_to_train (onnxruntime.capi._pybind_state.GraphInfo property)
(onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property)
input_names_require_grad (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property)
inputs (onnxruntime.capi._pybind_state.GradientNodeDefinition property)
insert() (onnxruntime.capi._pybind_state.OrtValueCache method)
inter_op_num_threads (onnxruntime.SessionOptions property)
intra_op_num_threads (onnxruntime.SessionOptions property)
io_binding() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
IOBinding (class in onnxruntime)
IODescription (class in onnxruntime.capi.ort_trainer)
is_sparse_tensor() (onnxruntime.OrtValue method)
is_tensor (onnxruntime.capi._pybind_state.GradientNodeAttributeDefinition property)
is_tensor() (onnxruntime.OrtValue method)
is_tensor_sequence() (onnxruntime.OrtValue method)
K
keys() (onnxruntime.capi._pybind_state.OrtValueCache method)
L
level (onnxruntime.capi._pybind_state.PropagateCastOpsConfiguration property)
load_checkpoint() (onnxruntime.training.ORTTrainer method)
load_state_dict() (onnxruntime.training.ORTModule method)
(onnxruntime.training.ORTTrainer 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)
loglevel (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property)
loss_output_name (onnxruntime.TrainingParameters property)
loss_scale (onnxruntime.TrainingParameters property)
loss_scale_input_name (onnxruntime.capi._pybind_state.TrainingConfigurationResult property)
lr_params_feed_name (onnxruntime.TrainingParameters property)
M
model_after_graph_transforms_path (onnxruntime.TrainingParameters property)
model_with_gradient_graph_path (onnxruntime.TrainingParameters property)
model_with_training_graph_path (onnxruntime.TrainingParameters property)
ModelDescription (class in onnxruntime.capi.ort_trainer)
module (onnxruntime.training.ORTModule property)
module_output_gradient_name (onnxruntime.capi._pybind_state.GraphInfo property)
module_output_indices_requires_save_for_backward (onnxruntime.capi._pybind_state.GraphInfo property)
modules() (onnxruntime.training.ORTModule method)
N
name (onnxruntime.capi._pybind_state.GradientNodeAttributeDefinition 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)
num_pipeline_micro_batches (onnxruntime.TrainingParameters property)
number_recompute_layers (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property)
(onnxruntime.TrainingParameters property)
numpy() (onnxruntime.OrtValue method)
O
only_execute_path_to_fetches (onnxruntime.RunOptions property)
op_type (onnxruntime.capi._pybind_state.GradientNodeDefinition property)
optimized_model_filepath (onnxruntime.SessionOptions property)
optimizer_attributes_map (onnxruntime.TrainingParameters property)
optimizer_int_attributes_map (onnxruntime.TrainingParameters property)
ort_value_from_sparse_tensor() (onnxruntime.OrtValue static method)
OrtDevice (class in onnxruntime)
ORTModule (class in onnxruntime.training)
OrtModuleGraphBuilder (class in onnxruntime.capi._pybind_state)
OrtModuleGraphBuilderConfiguration (class in onnxruntime.capi._pybind_state)
ORTTrainer (class in onnxruntime.capi.ort_trainer)
(class in onnxruntime.training)
ORTTrainerOptions (class in onnxruntime.training)
OrtValue (class in onnxruntime)
ortvalue_from_numpy() (onnxruntime.OrtValue static method)
ortvalue_from_shape_and_type() (onnxruntime.OrtValue static method)
OrtValueCache (class in onnxruntime.capi._pybind_state)
OrtValueVector (class in onnxruntime.capi._pybind_state)
output_grad_indices_non_differentiable (onnxruntime.capi._pybind_state.GraphInfo property)
output_grad_indices_require_full_shape (onnxruntime.capi._pybind_state.GraphInfo property)
outputs (onnxruntime.capi._pybind_state.GradientNodeDefinition property)
P
parameters() (onnxruntime.training.ORTModule method)
PartialGraphExecutionState (class in onnxruntime.capi._pybind_state)
pipeline_cut_info_string (onnxruntime.TrainingParameters property)
pipeline_parallel_size (onnxruntime.TrainingParameters property)
profile_file_prefix (onnxruntime.SessionOptions property)
propagate_cast_ops_allow (onnxruntime.TrainingParameters property)
propagate_cast_ops_config (onnxruntime.capi._pybind_state.GraphTransformerConfiguration property)
(onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property)
propagate_cast_ops_level (onnxruntime.TrainingParameters property)
PropagateCastOpsConfiguration (class in onnxruntime.capi._pybind_state)
push_back() (onnxruntime.capi._pybind_state.OrtValueVector method)
push_back_batch() (onnxruntime.capi._pybind_state.OrtValueVector method)
R
register_aten_op_executor() (in module onnxruntime.capi._pybind_state)
register_backward_hook() (onnxruntime.training.ORTModule method)
register_backward_runner() (in module onnxruntime.capi._pybind_state)
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_forward_runner() (in module onnxruntime.capi._pybind_state)
register_full_backward_hook() (onnxruntime.training.ORTModule method)
register_gradient_definition() (in module onnxruntime.capi._pybind_state)
register_module() (onnxruntime.training.ORTModule method)
register_parameter() (onnxruntime.training.ORTModule method)
register_torch_autograd_function() (in module onnxruntime.capi._pybind_state)
remove() (onnxruntime.capi._pybind_state.OrtValueCache method)
requires_grad_() (onnxruntime.training.ORTModule method)
reserve() (onnxruntime.capi._pybind_state.OrtValueVector method)
run() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
run_with_iobinding() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
run_with_ort_values() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
run_with_ortvaluevector() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
RunOptions (class in onnxruntime)
S
save_as_onnx() (onnxruntime.training.ORTTrainer method)
save_checkpoint() (onnxruntime.training.ORTTrainer method)
SessionOptions (class in onnxruntime)
set_extra_state() (onnxruntime.training.ORTModule method)
set_gradients_as_graph_outputs (onnxruntime.TrainingParameters property)
set_optimizer_initial_state() (onnxruntime.TrainingParameters method)
set_providers() (onnxruntime.InferenceSession method)
(onnxruntime.TrainingSession method)
shape() (onnxruntime.OrtValue method)
share_memory() (onnxruntime.training.ORTModule method)
shrink_to_fit() (onnxruntime.capi._pybind_state.OrtValueVector method)
sliced_axes (onnxruntime.TrainingParameters property)
sliced_schema (onnxruntime.TrainingParameters property)
sliced_tensor_names (onnxruntime.TrainingParameters property)
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)
(onnxruntime.training.ORTTrainer method)
strategy (onnxruntime.capi._pybind_state.PropagateCastOpsConfiguration property)
synchronize_execution_providers (onnxruntime.RunOptions property)
T
terminate (onnxruntime.RunOptions property)
to() (onnxruntime.training.ORTModule method)
to_cuda() (onnxruntime.SparseTensor method)
to_dlpacks() (onnxruntime.capi._pybind_state.OrtValueVector method)
to_empty() (onnxruntime.training.ORTModule method)
train() (onnxruntime.training.ORTModule method)
train_step() (onnxruntime.capi.ort_trainer.ORTTrainer method)
(onnxruntime.training.ORTTrainer method)
training_mode (onnxruntime.RunOptions property)
training_optimizer_name (onnxruntime.TrainingParameters property)
TrainingConfigurationResult (class in onnxruntime.capi._pybind_state)
TrainingGraphTransformerConfiguration (class in onnxruntime.capi._pybind_state)
TrainingParameters (class in onnxruntime)
TrainingSession (class in onnxruntime)
transformer_layer_recompute (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property)
(onnxruntime.TrainingParameters property)
type() (onnxruntime.training.ORTModule method)
U
unregister_python_functions() (in module onnxruntime.capi._pybind_state)
update_inplace() (onnxruntime.OrtValue method)
use_deterministic_compute (onnxruntime.SessionOptions property)
use_fp16_moments (onnxruntime.TrainingParameters property)
use_memory_efficient_gradient (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property)
(onnxruntime.TrainingParameters property)
use_mixed_precision (onnxruntime.TrainingParameters property)
user_input_grad_names (onnxruntime.capi._pybind_state.GraphInfo property)
user_input_names (onnxruntime.capi._pybind_state.GraphInfo property)
user_output_names (onnxruntime.capi._pybind_state.GraphInfo property)
V
value_json (onnxruntime.capi._pybind_state.GradientNodeAttributeDefinition property)
values() (onnxruntime.SparseTensor method)
W
weights_not_to_train (onnxruntime.TrainingParameters property)
weights_to_train (onnxruntime.TrainingParameters property)
world_rank (onnxruntime.TrainingParameters property)
world_size (onnxruntime.TrainingParameters property)
X
xpu() (onnxruntime.training.ORTModule method)
Z
zero_grad() (onnxruntime.training.ORTModule method)