Index Symbols | _ | A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Z Symbols 2021-10 2021-12 _ __annotations__ (onnxruntime.training.ortmodule.ORTModule attribute) __call__() (onnxruntime.training.ortmodule.ORTModule method) __class__ (onnxruntime.InferenceSession attribute) (onnxruntime.training.ortmodule.ORTModule attribute) (onnxruntime.TrainingSession attribute) __del__() (onnxruntime.TrainingSession method) __delattr__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __dir__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __dlpack__() (onnxruntime.capi._pybind_state.OrtValue method) __dlpack_device__() (onnxruntime.capi._pybind_state.OrtValue method) __eq__() (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __format__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __ge__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __getattr__() (onnxruntime.training.ortmodule.ORTModule method) __getattribute__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __getitem__() (onnxruntime.capi._pybind_state.OrtValueVector method) __getstate__() (onnxcustom.training._base_estimator.BaseEstimator method) (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) (onnxcustom.training.data_loader.OrtDataLoader method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) (onnxcustom.training.ortgradient.OrtGradientForwardBackward method) (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) (onnxruntime.training.ortmodule.ORTModule method) __gt__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __hash__() (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __index__() (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) __init__() (onnx.hub.ModelInfo method) (onnxcustom.training._base_estimator.BaseEstimator method) (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) (onnxcustom.training.data_loader.OrtDataLoader method) (onnxcustom.training.optimizers.OrtGradientOptimizer method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) (onnxcustom.training.ortgradient.OrtGradientForwardBackward method) (onnxcustom.training.ortgradient.OrtGradientForwardBackwardFunction method) (onnxcustom.training.sgd_learning_loss.AbsoluteLearningLoss method) (onnxcustom.training.sgd_learning_loss.BaseLearningLoss method) (onnxcustom.training.sgd_learning_loss.ElasticLearningLoss method) (onnxcustom.training.sgd_learning_loss.NegLogLearningLoss method) (onnxcustom.training.sgd_learning_loss.SquareLearningLoss method) (onnxcustom.training.sgd_learning_penalty.BaseLearningPenalty method) (onnxcustom.training.sgd_learning_penalty.ElasticLearningPenalty method) (onnxcustom.training.sgd_learning_penalty.NoLearningPenalty method) (onnxcustom.training.sgd_learning_rate.BaseLearningRate method) (onnxcustom.training.sgd_learning_rate.LearningRateSGD method) (onnxcustom.training.sgd_learning_rate.LearningRateSGDNesterov method) (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GradientGraphBuilder method) (onnxruntime.capi._pybind_state.GraphInfo method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.InferenceSession method) (onnxruntime.capi._pybind_state.ModelMetadata method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtDevice method) (onnxruntime.capi._pybind_state.OrtMemoryInfo method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method) (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.capi._pybind_state.OrtValueCache method) (onnxruntime.capi._pybind_state.OrtValueVector method) (onnxruntime.capi._pybind_state.PartialGraphExecutionState method) (onnxruntime.capi._pybind_state.RunOptions method) (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.capi._pybind_state.SessionOptions method) (onnxruntime.capi._pybind_state.SparseBlockSparseView method) (onnxruntime.capi._pybind_state.SparseCooView method) (onnxruntime.capi._pybind_state.SparseCsrView method) (onnxruntime.capi._pybind_state.SparseTensor method) (onnxruntime.capi._pybind_state.TrainingAgent method) (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration method) (onnxruntime.capi._pybind_state.TrainingParameters method) (onnxruntime.capi._pybind_state.TrainingSession method) (onnxruntime.InferenceSession method) (onnxruntime.OrtDevice method) (onnxruntime.OrtValue method) (onnxruntime.SessionIOBinding method) (onnxruntime.SparseTensor method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __init_subclass__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __int__() (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) __iter__() (onnxruntime.capi._pybind_state.OrtValueVector method) __le__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __len__() (onnxcustom.training.data_loader.OrtDataLoader method) (onnxruntime.capi._pybind_state.OrtValueVector method) __lt__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __members__ (onnxruntime.capi._pybind_state.ExecutionMode attribute) (onnxruntime.capi._pybind_state.ExecutionOrder attribute) (onnxruntime.capi._pybind_state.GraphOptimizationLevel attribute) (onnxruntime.capi._pybind_state.OrtAllocatorType attribute) (onnxruntime.capi._pybind_state.OrtMemType attribute) (onnxruntime.capi._pybind_state.OrtSparseFormat attribute) __ne__() (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __new__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __reduce__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __reduce_ex__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __repr__() (onnx.hub.ModelInfo method) (onnxcustom.training._base_estimator.BaseEstimator method) (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) (onnxcustom.training.data_loader.OrtDataLoader method) (onnxcustom.training.ortgradient.OrtGradientForwardBackward method) (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __repr_extended__() (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) (onnxcustom.training.sgd_learning_rate.BaseLearningRate method) __setattr__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __setstate__() (onnxcustom.training._base_estimator.BaseEstimator method) (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) (onnxcustom.training.data_loader.OrtDataLoader method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) (onnxcustom.training.ortgradient.OrtGradientForwardBackward method) (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) (onnxruntime.training.ortmodule.ORTModule method) __sizeof__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __slots__ (onnx.AttributeProto attribute) (onnx.FunctionProto attribute) (onnx.GraphProto attribute) (onnx.MapProto attribute) (onnx.ModelProto attribute) (onnx.NodeProto attribute) (onnx.SequenceProto attribute) (onnx.SparseTensorProto attribute) (onnx.TensorProto attribute) (onnx.TensorProto.Segment attribute) (onnx.TrainingInfoProto attribute) (onnx.ValueInfoProto attribute) __str__() (onnx.hub.ModelInfo method) (onnxruntime.capi._pybind_state.ExecutionMode method) (onnxruntime.capi._pybind_state.ExecutionOrder method) (onnxruntime.capi._pybind_state.GraphOptimizationLevel method) (onnxruntime.capi._pybind_state.OrtAllocatorType method) (onnxruntime.capi._pybind_state.OrtMemType method) (onnxruntime.capi._pybind_state.OrtSparseFormat method) (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __subclasshook__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) _apply() (onnxruntime.training.ortmodule.ORTModule method) _bind_input_ortvalue() (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) (onnxcustom.training.optimizers.OrtGradientOptimizer method) _bind_output_ortvalue() (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) _bio_cache() (onnxcustom.training._base_onnx_function.BaseLearningOnnx static method) _bio_do_bind_in() (onnxcustom.training._base_onnx_function.BaseLearningOnnx static method) _bio_do_bind_out() (onnxcustom.training._base_onnx_function.BaseLearningOnnx static method) _bio_ptr() (onnxcustom.training._base_onnx_function.BaseLearningOnnx static method) _cache_in_clear() (onnxcustom.training._base_onnx_function.BaseLearningOnnx static method) _call_impl() (onnxruntime.training.ortmodule.ORTModule method) _call_iobinding() (onnxcustom.training.sgd_learning_loss.BaseLearningLoss method) (onnxcustom.training.sgd_learning_penalty.BaseLearningPenalty method) (onnxcustom.training.sgd_learning_rate.BaseLearningRate method) _create_inference_session() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) _create_onnx_graphs() (onnxcustom.training.ortgradient.OrtGradientForwardBackward method) _create_training_session() (onnxcustom.training.optimizers.OrtGradientOptimizer method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) _default_inputs() (in module onnxcustom.training.grad_helper) _demangle() (in module onnxcustom.utils.nvprof2json) _evaluation() (onnxcustom.training.optimizers.OrtGradientOptimizer method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) _existing_names() (in module onnxcustom.utils.onnx_rewriter) _finalize_new_onnx() (in module onnxcustom.utils.onnx_helper) _get_att_state() (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) _get_backward_hooks() (onnxruntime.training.ortmodule.ORTModule method) _get_c_device() (onnxruntime.OrtDevice method) _get_c_tensor() (onnxruntime.SparseTensor method) _get_c_value() (onnxruntime.OrtValue method) _get_name() (onnxruntime.training.ortmodule.ORTModule method) _get_param_names() (onnxcustom.training._base.BaseOnnxClass class method) (onnxcustom.training._base_estimator.BaseEstimator class method) _get_trained_onnx() (onnxcustom.training._base_estimator.BaseEstimator method) _init_next() (onnxcustom.training.ortgradient.OrtGradientForwardBackward method) _is_full_backward_hook (onnxruntime.training.ortmodule.ORTModule attribute) _is_training() (onnxruntime.training.ortmodule.ORTModule method) _iteration() (onnxcustom.training.optimizers.OrtGradientOptimizer method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) _load_from_state_dict() (onnxruntime.training.ortmodule.ORTModule method) _loss_elastic() (in module onnxcustom.utils.orttraining_helper) _loss_l1() (in module onnxcustom.utils.orttraining_helper) _loss_l2() (in module onnxcustom.utils.orttraining_helper) _loss_log() (in module onnxcustom.utils.orttraining_helper) _maybe_warn_non_full_backward_hook() (onnxruntime.training.ortmodule.ORTModule method) _munge_time() (in module onnxcustom.utils.nvprof2json) _named_members() (onnxruntime.training.ortmodule.ORTModule method) _next_iter() (onnxcustom.training.data_loader.OrtDataLoader method) _onnx_axpy() (in module onnxcustom.utils.onnx_function) _onnx_axpyw() (in module onnxcustom.utils.onnx_function) _onnx_axpyw2() (in module onnxcustom.utils.onnx_function) _onnx_copy() (in module onnxcustom.utils.onnx_function) _onnx_derivative_fw() (in module onnxcustom.training.grad_helper) _onnx_derivative_loss() (in module onnxcustom.training.grad_helper) _onnx_grad_loss_absolute_error() (in module onnxcustom.utils.onnx_function) _onnx_grad_loss_elastic_error() (in module onnxcustom.utils.onnx_function) _onnx_grad_loss_square_error() (in module onnxcustom.utils.onnx_function) _onnx_grad_penalty_elastic_error() (in module onnxcustom.utils.onnx_function) _onnx_grad_sigmoid_neg_log_loss_error() (in module onnxcustom.utils.onnx_function) _onnx_grad_square_error() (in module onnxcustom.utils.onnx_function) _onnx_linear_regression() (in module onnxcustom.utils.onnx_function) _onnx_n_penalty_elastic_error() (in module onnxcustom.utils.onnx_function) _onnx_rewrite_operator_node() (in module onnxcustom.utils.onnx_rewriter) _onnx_square_error() (in module onnxcustom.utils.onnx_function) _onnx_update_penalty_elastic_error() (in module onnxcustom.utils.onnx_function) _onnx_zero() (in module onnxcustom.utils.onnx_function) _provider_name_to_device_type() (onnxcustom.training.ortgradient.OrtGradientForwardBackward static method) _register_load_state_dict_pre_hook() (onnxruntime.training.ortmodule.ORTModule method) _register_state_dict_hook() (onnxruntime.training.ortmodule.ORTModule method) _replicate_for_data_parallel() (onnxruntime.training.ortmodule.ORTModule method) _repr_helper_() (onnxcustom.training.ortgradient.OrtGradientForwardBackward static method) _reset_session() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) _rewrite_op_no_grad() (in module onnxcustom.utils.orttraining_helper) _save_to_state_dict() (onnxruntime.training.ortmodule.ORTModule method) _select_initializer_names() (onnxcustom.training.ortgradient.OrtGradientForwardBackward static method) _sizeof_fmt() (in module onnxcustom.utils.nvprof2json) _slow_forward() (onnxruntime.training.ortmodule.ORTModule method) _unique_name() (in module onnxcustom.utils.onnx_rewriter) (in module onnxcustom.utils.orttraining_helper) _version (onnxruntime.training.ortmodule.ORTModule attribute) A AbsoluteLearningLoss (class in onnxcustom.training.sgd_learning_loss) add_external_initializers() (onnxruntime.capi._pybind_state.SessionOptions method) add_free_dimension_override_by_denotation() (onnxruntime.capi._pybind_state.SessionOptions method) add_free_dimension_override_by_name() (onnxruntime.capi._pybind_state.SessionOptions method) add_initializer() (in module onnxcustom.utils.onnx_helper) (onnxruntime.capi._pybind_state.SessionOptions method) add_loss_output() (in module onnxcustom.utils.orttraining_helper) add_module() (onnxruntime.training.ortmodule.ORTModule method) add_prefix() (in module onnx.compose) add_prefix_graph() (in module onnx.compose) add_run_config_entry() (onnxruntime.capi._pybind_state.RunOptions method) add_session_config_entry() (onnxruntime.capi._pybind_state.SessionOptions method) algorithm (onnx.TrainingInfoProto attribute) allreduce_post_accumulation (onnxruntime.capi._pybind_state.TrainingParameters property) apply() (onnxruntime.training.ortmodule.ORTModule method) as_blocksparse_view() (onnxruntime.SparseTensor method) as_coo_view() (onnxruntime.SparseTensor method) as_csrc_view() (onnxruntime.SparseTensor method) as_sparse_tensor() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) attn_dropout_recompute (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property) (onnxruntime.capi._pybind_state.TrainingParameters property) attribute (onnx.FunctionProto attribute) (onnx.NodeProto attribute) AttributeProto (class in onnx) B backward() (onnxcustom.training.ortgradient.OrtGradientForwardBackwardFunction method) BaseEstimator (class in onnxcustom.training._base_estimator) BaseLearningLoss (class in onnxcustom.training.sgd_learning_loss) BaseLearningOnnx (class in onnxcustom.training._base_onnx_function) BaseLearningPenalty (class in onnxcustom.training.sgd_learning_penalty) BaseLearningRate (class in onnxcustom.training.sgd_learning_rate) BaseOnnxClass (class in onnxcustom.training._base) begin (onnx.TensorProto.Segment attribute) benchmark bfloat16() (onnxruntime.training.ortmodule.ORTModule method) bind_input() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) bind_ortvalue_input() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) bind_ortvalue_output() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) bind_output() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) black list blocksparse_from_numpy() (onnxruntime.capi._pybind_state.SparseTensor static method) blog, [1] bool_tensor_indices() (onnxruntime.capi._pybind_state.OrtValueVector method) buffers() (onnxruntime.training.ortmodule.ORTModule method) build build() (onnxruntime.capi._pybind_state.GradientGraphBuilder method) (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method) build_gradient_graph (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property) build_onnx_function() (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) (onnxcustom.training.sgd_learning_loss.AbsoluteLearningLoss method) (onnxcustom.training.sgd_learning_loss.ElasticLearningLoss method) (onnxcustom.training.sgd_learning_loss.NegLogLearningLoss method) (onnxcustom.training.sgd_learning_loss.SquareLearningLoss method) (onnxcustom.training.sgd_learning_penalty.ElasticLearningPenalty method) (onnxcustom.training.sgd_learning_penalty.NoLearningPenalty method) (onnxcustom.training.sgd_learning_rate.LearningRateSGD method) (onnxcustom.training.sgd_learning_rate.LearningRateSGDNesterov method) build_onnx_score_function() (onnxcustom.training.sgd_learning_loss.BaseLearningLoss method) C Cbids (class in onnxcustom.utils.nvprof2json) check check() (in module onnxcustom) check_model() (in module onnx.checker) children() (onnxruntime.training.ortmodule.ORTModule method) clear() (onnxruntime.capi._pybind_state.OrtValueCache method) clear_binding_inputs() (onnxcustom.training._base_onnx_function.BaseLearningOnnx method) (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) clear_binding_outputs() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) clear_training_ep_instances() (in module onnxruntime.capi._pybind_state) combine_pairs_to_complex() (in module onnx.numpy_helper) ConvergenceError ConvergenceWarning convert_endian() (in module onnx.numpy_helper) convert_trace_to_json() (in module onnxcustom.utils.nvprof2json) convert_version() (in module onnx.version_converter) Converts a profile stored by nvprof into json copy_outputs_to_cpu() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) count() (onnxruntime.capi._pybind_state.OrtValueCache method) cpu() (onnxruntime.capi._pybind_state.OrtDevice static method) (onnxruntime.training.ortmodule.ORTModule method) create_and_register_allocator() (in module onnxruntime.capi._pybind_state) cuda() (onnxruntime.capi._pybind_state.OrtDevice static method) (onnxruntime.training.ortmodule.ORTModule method) custom converter, [1] custom python runtime custom_metadata_map (onnxruntime.capi._pybind_state.ModelMetadata property) D data_location (onnx.TensorProto attribute) data_np (onnxcustom.training.data_loader.OrtDataLoader property) data_ort (onnxcustom.training.data_loader.OrtDataLoader property) data_parallel_size (onnxruntime.capi._pybind_state.TrainingParameters property) data_ptr() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) data_type (onnx.TensorProto attribute) data_type() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.capi._pybind_state.SparseTensor method) (onnxruntime.OrtValue method) (onnxruntime.SparseTensor method) dataframe, [1] deep learning deepspeed_zero_stage (onnxruntime.capi._pybind_state.TrainingParameters property) default_memory() (onnxruntime.capi._pybind_state.OrtDevice static method) dense_shape() (onnxruntime.capi._pybind_state.SparseTensor method) (onnxruntime.SparseTensor method) deployment DerivativeOptions (class in onnxcustom.training.grad_helper) description (onnxruntime.capi._pybind_state.ModelMetadata property) DESCRIPTOR (onnx.AttributeProto attribute) (onnx.FunctionProto attribute) (onnx.GraphProto attribute) (onnx.MapProto attribute) (onnx.ModelProto attribute) (onnx.NodeProto attribute) (onnx.SequenceProto attribute) (onnx.SparseTensorProto attribute) (onnx.TensorProto attribute) (onnx.TensorProto.Segment attribute) (onnx.TrainingInfoProto attribute) (onnx.ValueInfoProto attribute) device_id() (onnxruntime.capi._pybind_state.OrtDevice method) (onnxruntime.OrtDevice method) device_name() (onnxcustom.training.ortgradient.OrtGradientForwardBackwardFunction static method) (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.capi._pybind_state.SparseTensor method) (onnxruntime.OrtValue method) (onnxruntime.SparseTensor method) device_to_providers() (in module onnxcustom.utils.onnxruntime_helper) device_type() (onnxruntime.capi._pybind_state.OrtDevice method) (onnxruntime.OrtDevice method) dims (onnx.SparseTensorProto attribute) (onnx.TensorProto attribute) disable_fallback() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) disable_telemetry_events() (in module onnxruntime.capi._pybind_state) discrepencies, [1] dlpack_at() (onnxruntime.capi._pybind_state.OrtValueVector method) doc_string (onnx.AttributeProto attribute) (onnx.FunctionProto attribute) (onnx.GraphProto attribute) (onnx.ModelProto attribute) (onnx.NodeProto attribute) (onnx.TensorProto attribute) (onnx.ValueInfoProto attribute) domain (onnx.FunctionProto attribute) (onnx.ModelProto attribute) (onnx.NodeProto attribute) (onnxruntime.capi._pybind_state.ModelMetadata property) double double() (onnxruntime.training.ortmodule.ORTModule method) double_data (onnx.TensorProto attribute) dtype_to_var_type() (in module onnxcustom.utils.onnx_helper) dump_patches (onnxruntime.training.ortmodule.ORTModule attribute) E ElasticLearningLoss (class in onnxcustom.training.sgd_learning_loss) ElasticLearningPenalty (class in onnxcustom.training.sgd_learning_penalty) elem_type (onnx.SequenceProto attribute) element_type() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) element_type_at() (onnxruntime.capi._pybind_state.OrtValueVector method) enable_adasum (onnxruntime.capi._pybind_state.TrainingParameters property) enable_caching (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property) enable_cpu_mem_arena (onnxruntime.capi._pybind_state.SessionOptions property) enable_fallback() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) enable_gelu_approximation (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property) enable_grad_norm_clip (onnxruntime.capi._pybind_state.TrainingParameters property) enable_mem_pattern (onnxruntime.capi._pybind_state.SessionOptions property) enable_mem_reuse (onnxruntime.capi._pybind_state.SessionOptions property) enable_profiling (onnxruntime.capi._pybind_state.SessionOptions property) enable_telemetry_events() (in module onnxruntime.capi._pybind_state) end (onnx.TensorProto.Segment attribute) end_profiling() (onnxruntime.capi._pybind_state.InferenceSession method) (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) EngineError (class in onnxruntime.capi._pybind_state) ensemble EPFail (class in onnxruntime.capi._pybind_state) eval() (onnxruntime.training.ortmodule.ORTModule method) EvaluationError execution_mode (onnxruntime.capi._pybind_state.SessionOptions property) execution_order (onnxruntime.capi._pybind_state.SessionOptions property) ExecutionMode (class in onnxruntime.capi._pybind_state) ExecutionOrder (class in onnxruntime.capi._pybind_state) expand_out_dim() (in module onnx.compose) expand_out_dim_graph() (in module onnx.compose) export_gradient_graph() (in module onnxruntime.training.experimental.gradient_graph._gradient_graph_tools) external_data (onnx.TensorProto attribute) extra_repr() (onnxruntime.training.ortmodule.ORTModule method) extract_model() (in module onnx.utils) F f (onnx.AttributeProto attribute) Fail (class in onnxruntime.capi._pybind_state) finalize() (onnxruntime.capi._pybind_state.TrainingSession method) find_min_ir_version_for() (in module onnx.helper) fit() (onnxcustom.training.optimizers.OrtGradientOptimizer method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) fix_link_operator_md() (in module onnxcustom.utils.doc_helper) float float() (onnxruntime.training.ortmodule.ORTModule method) float_data (onnx.TensorProto attribute) floats (onnx.AttributeProto attribute) format (onnxruntime.capi._pybind_state.SparseTensor property) format() (onnxruntime.SparseTensor method) forward() (onnxcustom.training.ortgradient.OrtGradientForwardBackwardFunction method) (onnxruntime.training.ortmodule.ORTModule method) from_array() (in module onnx.numpy_helper) from_dict() (in module onnx.numpy_helper) from_dlpack() (onnxruntime.capi._pybind_state.OrtValue static method) from_list() (in module onnx.numpy_helper) from_optional() (in module onnx.numpy_helper) function_onnx_graph() (in module onnxcustom.utils.onnx_function) FunctionProto (class in onnx) functions (onnx.ModelProto attribute) G g (onnx.AttributeProto attribute) gelu_recompute (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property) (onnxruntime.capi._pybind_state.TrainingParameters property) get_all_providers() (in module onnxruntime) get_all_schemas() (in module onnx.onnx_cpp2py_export.defs) get_all_schemas_with_history() (in module onnx.defs) (in module onnx.onnx_cpp2py_export.defs) get_attribute_value() (in module onnx.helper) get_available_providers() (in module onnxruntime) get_blocksparse_data() (onnxruntime.capi._pybind_state.SparseTensor method) get_buffer() (onnxruntime.training.ortmodule.ORTModule method) get_class_names() (in module onnxcustom.utils.imagenet_classes) get_coo_data() (onnxruntime.capi._pybind_state.SparseTensor method) get_csrc_data() (onnxruntime.capi._pybind_state.SparseTensor method) get_device() (in module onnxruntime) get_extra_state() (onnxruntime.training.ortmodule.ORTModule method) get_full_state() (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) get_function_ops() (in module onnx.defs) get_graph_info() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method) get_inference_optimized_model() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method) get_initializer() (onnxcustom.training.ortgradient.OrtGradientForwardBackward method) get_inputs() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) get_max_opset() (in module onnxcustom) get_model() (onnxruntime.capi._pybind_state.GradientGraphBuilder method) (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method) get_model_info() (in module onnx.hub) get_model_state() (onnxruntime.capi._pybind_state.TrainingSession method) (onnxruntime.TrainingSession method) get_modelmeta() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) get_onnx_opset() (in module onnxcustom.utils.onnx_helper) get_optimizer_state() (onnxruntime.capi._pybind_state.TrainingSession method) (onnxruntime.TrainingSession method) get_ort_device() (in module onnxcustom.utils.onnxruntime_helper) get_ort_device_type() (in module onnxcustom.utils.onnxruntime_helper) get_outputs() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.InferenceSession method) (onnxruntime.SessionIOBinding method) (onnxruntime.TrainingSession method) get_overridable_initializers() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) get_parameter() (onnxruntime.training.ortmodule.ORTModule method) get_params() (onnxcustom.training._base_estimator.BaseEstimator method) get_partition_info_map() (onnxruntime.capi._pybind_state.TrainingSession method) (onnxruntime.TrainingSession method) get_profiling_start_time_ns (onnxruntime.capi._pybind_state.InferenceSession property) get_profiling_start_time_ns() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) get_provider_options() (onnxruntime.capi._pybind_state.InferenceSession method) (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) get_providers() (onnxruntime.capi._pybind_state.InferenceSession method) (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) get_run_config_entry() (onnxruntime.capi._pybind_state.RunOptions method) get_schema() (in module onnx.defs) (in module onnx.onnx_cpp2py_export.defs) get_session_config_entry() (onnxruntime.capi._pybind_state.SessionOptions method) get_session_initializer() (in module onnxruntime.capi._pybind_state) get_session_options() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) get_state() (onnxcustom.training.optimizers.OrtGradientOptimizer method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) (onnxruntime.capi._pybind_state.TrainingSession method) (onnxruntime.TrainingSession method) get_submodule() (onnxruntime.training.ortmodule.ORTModule method) get_supported_functions() (in module onnxcustom.utils.onnx_function) get_train_initializer() (in module onnxcustom.utils.orttraining_helper) get_trained_onnx() (onnxcustom.training._base_estimator.BaseEstimator method) (onnxcustom.training.optimizers.OrtGradientOptimizer method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) GetOpNodeProducer() (in module onnx.tools.net_drawer) GetPydotGraph() (in module onnx.tools.net_drawer) gradient_accumulation_steps (onnxruntime.capi._pybind_state.TrainingParameters property) GradientGraphBuilder (class in onnxruntime.capi._pybind_state), [1] GradientNodeAttributeDefinition (class in onnxruntime.capi._pybind_state) GradientNodeDefinition (class in onnxruntime.capi._pybind_state) graph (onnx.ModelProto attribute) graph_description (onnxruntime.capi._pybind_state.ModelMetadata property) graph_name (onnxruntime.capi._pybind_state.ModelMetadata property) graph_optimization_level (onnxruntime.capi._pybind_state.SessionOptions property) graph_transformer_config (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property) GraphInfo (class in onnxruntime.capi._pybind_state) GraphOptimizationLevel (class in onnxruntime.capi._pybind_state) GraphProto (class in onnx) graphs (onnx.AttributeProto attribute) H half() (onnxruntime.training.ortmodule.ORTModule method) has_schema() (in module onnx.onnx_cpp2py_export.defs) has_value() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) have_torch() (in module onnxruntime.tools) horizontal_parallel_size (onnxruntime.capi._pybind_state.TrainingParameters property) I i (onnx.AttributeProto attribute) immutable_weights (onnxruntime.capi._pybind_state.TrainingParameters property) incomplete, [1], [2] indices (onnx.SparseTensorProto attribute) indices() (onnxruntime.capi._pybind_state.SparseBlockSparseView method) (onnxruntime.capi._pybind_state.SparseCooView method) infer_input_info() (in module onnxruntime.tools) infer_shapes() (in module onnx.shape_inference) infer_shapes_path() (in module onnx.shape_inference) InferenceSession (class in onnxruntime) (class in onnxruntime.capi._pybind_state) init_learning_rate() (onnxcustom.training.sgd_learning_rate.BaseLearningRate method) (onnxcustom.training.sgd_learning_rate.LearningRateSGD method) (onnxcustom.training.sgd_learning_rate.LearningRateSGDNesterov method) initialization (onnx.TrainingInfoProto attribute) initialization_binding (onnx.TrainingInfoProto attribute) initialize() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method) initialize_session() (onnxruntime.capi._pybind_state.InferenceSession method) initializer (onnx.GraphProto attribute) initializer_names (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property) initializer_names_to_train (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property) inner() (onnxruntime.capi._pybind_state.SparseCsrView method) input (onnx.FunctionProto attribute) (onnx.GraphProto attribute) (onnx.NodeProto attribute) input_names_require_grad (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property) input_to_ort() (onnxcustom.training.ortgradient.OrtGradientForwardBackwardFunction static method) inputs_meta (onnxruntime.capi._pybind_state.InferenceSession property) insert() (onnxruntime.capi._pybind_state.OrtValueCache method) int32_data (onnx.TensorProto attribute) int64_data (onnx.TensorProto attribute) inter_op_num_threads (onnxruntime.capi._pybind_state.SessionOptions property) intermediate results intra_op_num_threads (onnxruntime.capi._pybind_state.SessionOptions property) ints (onnx.AttributeProto attribute) InvalidArgument (class in onnxruntime.capi._pybind_state) InvalidGraph (class in onnxruntime.capi._pybind_state) InvalidProtobuf (class in onnxruntime.capi._pybind_state) investigate io_binding() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) ir_version (onnx.ModelProto attribute) is_dlpack_uint8_tensor() (in module onnxruntime.capi._pybind_state) is_output_fp32_node() (onnxruntime.capi._pybind_state.TrainingSession method) (onnxruntime.TrainingSession method) is_sparse_tensor() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) is_tensor() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) is_tensor_sequence() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) iter_bind() (onnxcustom.training.data_loader.OrtDataLoader method) iter_numpy() (onnxcustom.training.data_loader.OrtDataLoader method) iter_ortvalue() (onnxcustom.training.data_loader.OrtDataLoader method) J json_to_dataframe() (in module onnxcustom.utils.nvprof2json) json_to_dataframe_streaming() (in module onnxcustom.utils.nvprof2json) K key_type (onnx.MapProto attribute) keys (onnx.MapProto attribute) keys() (onnxruntime.capi._pybind_state.OrtValueCache method) L LearningRateSGD (class in onnxcustom.training.sgd_learning_rate) LearningRateSGDNesterov (class in onnxcustom.training.sgd_learning_rate) LightGBM, [1] lightgbm list_models() (in module onnx.hub) load() (in module onnx) (in module onnx.hub) load_model() (onnxruntime.capi._pybind_state.TrainingSession method) load_model_from_string() (in module onnx) load_state() (onnxruntime.capi._pybind_state.TrainingSession method) (onnxruntime.TrainingSession method) load_state_dict() (onnxruntime.training.ortmodule.ORTModule method) load_tensor_from_string() (in module onnx) log_severity_level (onnxruntime.capi._pybind_state.RunOptions property) (onnxruntime.capi._pybind_state.SessionOptions property) log_verbosity_level (onnxruntime.capi._pybind_state.RunOptions property) (onnxruntime.capi._pybind_state.SessionOptions property) logid (onnxruntime.capi._pybind_state.RunOptions property) (onnxruntime.capi._pybind_state.SessionOptions property) loglevel (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property) loop() (onnxcustom.training.sgd_learning_rate.BaseLearningRate method) loss_gradient() (onnxcustom.training.sgd_learning_loss.BaseLearningLoss method) loss_output_name (onnxruntime.capi._pybind_state.TrainingParameters property) loss_scale (onnxruntime.capi._pybind_state.TrainingParameters property) loss_scores() (onnxcustom.training.sgd_learning_loss.BaseLearningLoss method) losses() (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) lr_params_feed_name (onnxruntime.capi._pybind_state.TrainingParameters property) M main() (in module onnxcustom.__main__) make() (onnxruntime.OrtDevice static method) make_attribute() (in module onnx.helper) make_empty_tensor_value_info() (in module onnx.helper) make_function() (in module onnx.helper) make_graph() (in module onnx.helper) make_map() (in module onnx.helper) make_model() (in module onnx.helper) make_node() (in module onnx.helper) make_operatorsetid() (in module onnx.helper) make_opsetid() (in module onnx.helper) make_optional() (in module onnx.helper) make_optional_type_proto() (in module onnx.helper) make_sequence() (in module onnx.helper) make_sequence_type_proto() (in module onnx.helper) make_sequence_value_info() (in module onnx.helper) make_sparse_tensor() (in module onnx.helper) make_sparse_tensor_type_proto() (in module onnx.helper) make_sparse_tensor_value_info() (in module onnx.helper) make_tensor() (in module onnx.helper) make_tensor_sequence_value_info() (in module onnx.helper) make_tensor_type_proto() (in module onnx.helper), [1] make_tensor_value_info() (in module onnx.helper) make_training_info() (in module onnx.helper) make_value_info() (in module onnx.helper) map_values (onnx.SequenceProto attribute) MapProto (class in onnx) measure_time() (in module onnxcustom.utils.benchmark) merge_graphs() (in module onnx.compose) merge_models() (in module onnx.compose) metadata (onnx.hub.ModelInfo attribute) metadata_props (onnx.ModelProto attribute) model (onnx.hub.ModelInfo attribute) model_after_graph_transforms_path (onnxruntime.capi._pybind_state.TrainingParameters property) model_meta (onnxruntime.capi._pybind_state.InferenceSession property) model_path (onnx.hub.ModelInfo attribute) model_sha (onnx.hub.ModelInfo attribute) model_version (onnx.ModelProto attribute) model_with_gradient_graph_path (onnxruntime.capi._pybind_state.TrainingParameters property) model_with_training_graph_path (onnxruntime.capi._pybind_state.TrainingParameters property) ModelInfo (class in onnx.hub) ModelLoaded (class in onnxruntime.capi._pybind_state) ModelMetadata (class in onnxruntime.capi._pybind_state) ModelProto (class in onnx), [1] module onnx.onnx_cpp2py_export.defs onnxcustom onnxcustom.__main__ onnxcustom.cli onnxcustom.cli.profiling onnxcustom.plotting onnxcustom.plotting.plotting_onnx onnxcustom.training onnxcustom.training._base onnxcustom.training._base_estimator onnxcustom.training._base_onnx_function onnxcustom.training.data_loader onnxcustom.training.excs onnxcustom.training.grad_helper onnxcustom.training.optimizers onnxcustom.training.optimizers_partial onnxcustom.training.ortgradient onnxcustom.training.sgd_learning_loss onnxcustom.training.sgd_learning_penalty onnxcustom.training.sgd_learning_rate onnxcustom.utils onnxcustom.utils.benchmark onnxcustom.utils.doc_helper onnxcustom.utils.imagenet_classes onnxcustom.utils.nvprof2json onnxcustom.utils.onnx_function onnxcustom.utils.onnx_helper onnxcustom.utils.onnx_rewriter onnxcustom.utils.onnxruntime_helper onnxcustom.utils.orttraining_helper onnxcustom.utils.print_helper module (onnxruntime.training.ortmodule.ORTModule property) modules() (onnxruntime.training.ortmodule.ORTModule method) N name (onnx.AttributeProto attribute) (onnx.FunctionProto attribute) (onnx.GraphProto attribute) (onnx.MapProto attribute) (onnx.NodeProto attribute) (onnx.SequenceProto attribute) (onnx.TensorProto attribute) (onnx.ValueInfoProto attribute) (onnxruntime.capi._pybind_state.ExecutionMode property) (onnxruntime.capi._pybind_state.ExecutionOrder property) (onnxruntime.capi._pybind_state.GraphOptimizationLevel property) (onnxruntime.capi._pybind_state.OrtAllocatorType property) (onnxruntime.capi._pybind_state.OrtMemType property) (onnxruntime.capi._pybind_state.OrtSparseFormat property) named_buffers() (onnxruntime.training.ortmodule.ORTModule method) named_children() (onnxruntime.training.ortmodule.ORTModule method) named_modules() (onnxruntime.training.ortmodule.ORTModule method) named_parameters() (onnxruntime.training.ortmodule.ORTModule method) needs_grad (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer property) (onnxcustom.training.sgd_learning_rate.BaseLearningRate property) (onnxcustom.training.sgd_learning_rate.LearningRateSGD property) (onnxcustom.training.sgd_learning_rate.LearningRateSGDNesterov property) NegLogLearningLoss (class in onnxcustom.training.sgd_learning_loss) new_instance() (onnxcustom.training.ortgradient.OrtGradientForwardBackward method) node (onnx.FunctionProto attribute) (onnx.GraphProto attribute) NodeProto (class in onnx) NoLearningPenalty (class in onnxcustom.training.sgd_learning_penalty) NoModel (class in onnxruntime.capi._pybind_state) NoSuchFile (class in onnxruntime.capi._pybind_state) NotImplemented (class in onnxruntime.capi._pybind_state) num_pipeline_micro_batches (onnxruntime.capi._pybind_state.TrainingParameters property) number_recompute_layers (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property) (onnxruntime.capi._pybind_state.TrainingParameters property) numpy() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) numpy_to_ort_value() (in module onnxcustom.utils.onnxruntime_helper) nvprof2json() (in module onnxcustom.cli.profiling) O only_execute_path_to_fetches (onnxruntime.capi._pybind_state.RunOptions property) ONNX graph onnx.onnx_cpp2py_export.defs module onnx_derivative() (in module onnxcustom.training.grad_helper) onnx_opset_version() (in module onnx.defs), [1] onnx_rename_weights() (in module onnxcustom.utils.onnx_helper) onnx_rewrite_operator() (in module onnxcustom.utils.onnx_rewriter) onnxcustom module onnxcustom.__main__ module onnxcustom.cli module onnxcustom.cli.profiling module onnxcustom.plotting module onnxcustom.plotting.plotting_onnx module onnxcustom.training module onnxcustom.training._base module onnxcustom.training._base_estimator module onnxcustom.training._base_onnx_function module onnxcustom.training.data_loader module onnxcustom.training.excs module onnxcustom.training.grad_helper module onnxcustom.training.optimizers module onnxcustom.training.optimizers_partial module onnxcustom.training.ortgradient module onnxcustom.training.sgd_learning_loss module onnxcustom.training.sgd_learning_penalty module onnxcustom.training.sgd_learning_rate module onnxcustom.utils module onnxcustom.utils.benchmark module onnxcustom.utils.doc_helper module onnxcustom.utils.imagenet_classes module onnxcustom.utils.nvprof2json module onnxcustom.utils.onnx_function module onnxcustom.utils.onnx_helper module onnxcustom.utils.onnx_rewriter module onnxcustom.utils.onnxruntime_helper module onnxcustom.utils.orttraining_helper module onnxcustom.utils.print_helper module op_type (onnx.NodeProto attribute) opset (onnx.hub.ModelInfo attribute) opset_import (onnx.FunctionProto attribute) (onnx.ModelProto attribute) optimize_model() (in module onnxruntime.tools.optimize_onnx_model) optimized_model_filepath (onnxruntime.capi._pybind_state.SessionOptions property) optimizer_attributes_map (onnxruntime.capi._pybind_state.TrainingParameters property) optimizer_int_attributes_map (onnxruntime.capi._pybind_state.TrainingParameters property) optional_values (onnx.SequenceProto attribute) options, [1] ort_device_to_string() (in module onnxcustom.utils.onnxruntime_helper) ort_value_from_sparse_tensor() (onnxruntime.capi._pybind_state.OrtValue static method) (onnxruntime.OrtValue static method) OrtAllocatorType (class in onnxruntime.capi._pybind_state) OrtDataLoader (class in onnxcustom.training.data_loader) OrtDevice (class in onnxruntime) (class in onnxruntime.capi._pybind_state) OrtGradientForwardBackward (class in onnxcustom.training.ortgradient) OrtGradientForwardBackwardFunction (class in onnxcustom.training.ortgradient) OrtGradientForwardBackwardOptimizer (class in onnxcustom.training.optimizers_partial) OrtGradientOptimizer (class in onnxcustom.training.optimizers) OrtMemoryInfo (class in onnxruntime.capi._pybind_state) OrtMemType (class in onnxruntime.capi._pybind_state) ORTModule (class in onnxruntime.training.ortmodule) OrtModuleGraphBuilder (class in onnxruntime.capi._pybind_state) OrtModuleGraphBuilderConfiguration (class in onnxruntime.capi._pybind_state) OrtSparseFormat (class in onnxruntime.capi._pybind_state) OrtValue (class in onnxruntime) (class in onnxruntime.capi._pybind_state) ortvalue_from_numpy() (onnxruntime.capi._pybind_state.OrtValue static method) (onnxruntime.OrtValue static method) ortvalue_from_shape_and_type() (onnxruntime.capi._pybind_state.OrtValue static method) (onnxruntime.OrtValue static method) OrtValueCache (class in onnxruntime.capi._pybind_state) OrtValueVector (class in onnxruntime.capi._pybind_state) outer() (onnxruntime.capi._pybind_state.SparseCsrView method) output (onnx.FunctionProto attribute) (onnx.GraphProto attribute) (onnx.NodeProto attribute) outputs_meta (onnxruntime.capi._pybind_state.InferenceSession property) overridable_initializers (onnxruntime.capi._pybind_state.InferenceSession property) P parameters() (onnxruntime.training.ortmodule.ORTModule method) parser, [1] PartialGraphExecutionState (class in onnxruntime.capi._pybind_state) penalty_loss() (onnxcustom.training.sgd_learning_penalty.BaseLearningPenalty method) (onnxcustom.training.sgd_learning_penalty.ElasticLearningPenalty method) (onnxcustom.training.sgd_learning_penalty.NoLearningPenalty method) penalty_loss_onnx() (in module onnxcustom.utils.orttraining_helper) pipeline pipeline_cut_info_string (onnxruntime.capi._pybind_state.TrainingParameters property) pipeline_parallel_size (onnxruntime.capi._pybind_state.TrainingParameters property) plot_onnxs() (in module onnxcustom.plotting.plotting_onnx) printable_attribute() (in module onnx.helper) printable_dim() (in module onnx.helper) printable_graph() (in module onnx.helper) printable_node() (in module onnx.helper) printable_tensor_proto() (in module onnx.helper) printable_type() (in module onnx.helper) printable_value_info() (in module onnx.helper) producer_name (onnx.ModelProto attribute) (onnxruntime.capi._pybind_state.ModelMetadata property) producer_version (onnx.ModelProto attribute) profile_file_prefix (onnxruntime.capi._pybind_state.SessionOptions property) propagate_cast_ops_allow (onnxruntime.capi._pybind_state.TrainingParameters property) propagate_cast_ops_config (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property) propagate_cast_ops_level (onnxruntime.capi._pybind_state.TrainingParameters property) proto_type_to_dtype() (in module onnxcustom.utils.onnx_helper) provider_to_device() (in module onnxcustom.utils.onnxruntime_helper) ProviderError push_back() (onnxruntime.capi._pybind_state.OrtValueVector method) pytorch_export_helpers() (in module onnxruntime.tools) Q quantization_annotation (onnx.GraphProto attribute) R raw_data (onnx.TensorProto attribute) read_bytes() (onnxruntime.capi._pybind_state.TrainingSession method) ref_attr_name (onnx.AttributeProto attribute) register_aten_op_executor() (in module onnxruntime.capi._pybind_state) register_backward_hook() (onnxruntime.training.ortmodule.ORTModule method) register_backward_runner() (in module onnxruntime.capi._pybind_state) register_buffer() (onnxruntime.training.ortmodule.ORTModule method) register_custom_ops_library() (onnxruntime.capi._pybind_state.SessionOptions method) register_forward_hook() (onnxruntime.training.ortmodule.ORTModule method) register_forward_pre_hook() (onnxruntime.training.ortmodule.ORTModule method) register_forward_runner() (in module onnxruntime.capi._pybind_state) register_full_backward_hook() (onnxruntime.training.ortmodule.ORTModule method) register_gradient_definition() (in module onnxruntime.capi._pybind_state) register_module() (onnxruntime.training.ortmodule.ORTModule method) register_parameter() (onnxruntime.training.ortmodule.ORTModule method) remove() (onnxruntime.capi._pybind_state.OrtValueCache method) replace_initializers_into_onnx() (in module onnxcustom.utils.onnx_helper) requires_grad_() (onnxruntime.training.ortmodule.ORTModule method) reserve() (onnxruntime.capi._pybind_state.OrtValueVector method) run() (onnxruntime.capi._pybind_state.InferenceSession method) (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) run_backward() (onnxruntime.capi._pybind_state.TrainingAgent method) run_forward() (onnxruntime.capi._pybind_state.TrainingAgent method) run_with_iobinding() (onnxruntime.capi._pybind_state.InferenceSession method) (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) run_with_ort_values() (onnxruntime.capi._pybind_state.InferenceSession method) (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) RunOptions (class in onnxruntime.capi._pybind_state) RuntimeException (class in onnxruntime.capi._pybind_state) S s (onnx.AttributeProto attribute) save() (onnxruntime.capi._pybind_state.GradientGraphBuilder method) save_for_backward() (onnxcustom.training.ortgradient.OrtGradientForwardBackwardFunction method) save_onnx_graph() (onnxcustom.training._base.BaseOnnxClass method) (onnxcustom.training.ortgradient.OrtGradientForwardBackwardFunction class method) saved_tensors (onnxcustom.training.ortgradient.OrtGradientForwardBackwardFunction property) schema_version_map() (in module onnx.onnx_cpp2py_export.defs) SchemaError score() (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) segment (onnx.TensorProto attribute) select() (onnxcustom.training.sgd_learning_loss.BaseLearningLoss static method) (onnxcustom.training.sgd_learning_penalty.BaseLearningPenalty static method) (onnxcustom.training.sgd_learning_rate.BaseLearningRate static method) sequence_values (onnx.SequenceProto attribute) SequenceProto (class in onnx) SerializeToString() (onnx.ModelProto method) (onnx.TensorProto method) session_options (onnxruntime.capi._pybind_state.InferenceSession property) SessionIOBinding (class in onnxruntime) (class in onnxruntime.capi._pybind_state) SessionOptions (class in onnxruntime.capi._pybind_state) set_default_logger_severity() (in module onnxruntime) set_extra_state() (onnxruntime.training.ortmodule.ORTModule method) set_gradients_as_graph_outputs (onnxruntime.capi._pybind_state.TrainingParameters property) set_optimizer_initial_state() (onnxruntime.capi._pybind_state.TrainingParameters method) set_params() (onnxcustom.training._base_estimator.BaseEstimator method) set_providers() (onnxruntime.InferenceSession method) (onnxruntime.TrainingSession method) set_seed() (in module onnxruntime) set_state() (onnxcustom.training.optimizers.OrtGradientOptimizer method) (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) shape() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) share_memory() (onnxruntime.training.ortmodule.ORTModule method) shrink_to_fit() (onnxruntime.capi._pybind_state.OrtValueVector method) sigmoid sliced_axes (onnxruntime.capi._pybind_state.TrainingParameters property) sliced_schema (onnxruntime.capi._pybind_state.TrainingParameters property) sliced_tensor_names (onnxruntime.capi._pybind_state.TrainingParameters property) sparse sparse_coo_from_numpy() (onnxruntime.capi._pybind_state.SparseTensor static method) (onnxruntime.SparseTensor static method) sparse_csr_from_numpy() (onnxruntime.capi._pybind_state.SparseTensor static method) (onnxruntime.SparseTensor static method) sparse_initializer (onnx.GraphProto attribute) sparse_tensor (onnx.AttributeProto attribute) sparse_tensor_values (onnx.SequenceProto attribute) sparse_tensors (onnx.AttributeProto attribute) SparseBlockSparseView (class in onnxruntime.capi._pybind_state) SparseCooView (class in onnxruntime.capi._pybind_state) SparseCsrView (class in onnxruntime.capi._pybind_state) SparseTensor (class in onnxruntime) (class in onnxruntime.capi._pybind_state) SparseTensorProto (class in onnx) split_complex_to_pairs() (in module onnx.helper) SquareLearningLoss (class in onnxcustom.training.sgd_learning_loss) state_dict() (onnxruntime.training.ortmodule.ORTModule method) str_ortvalue() (in module onnxcustom.utils.print_helper) string_data (onnx.TensorProto attribute) string_keys (onnx.MapProto attribute) strings (onnx.AttributeProto attribute) synchronize_inputs() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) synchronize_outputs() (onnxruntime.capi._pybind_state.SessionIOBinding method) (onnxruntime.SessionIOBinding method) syntax T t (onnx.AttributeProto attribute) tags (onnx.hub.ModelInfo attribute) target opset tensor_values (onnx.SequenceProto attribute) TensorProto (class in onnx), [1] TensorProto.Segment (class in onnx) tensors (onnx.AttributeProto attribute) terminate (onnxruntime.capi._pybind_state.RunOptions property) to() (onnxruntime.training.ortmodule.ORTModule method) to_array() (in module onnx.numpy_helper) to_cuda() (onnxruntime.capi._pybind_state.SparseTensor method) (onnxruntime.SparseTensor method) to_dict() (in module onnx.numpy_helper) to_dlpack() (onnxruntime.capi._pybind_state.OrtValue method) to_dlpacks() (onnxruntime.capi._pybind_state.OrtValueVector method) to_empty() (onnxruntime.training.ortmodule.ORTModule method) to_list() (in module onnx.numpy_helper) to_optional() (in module onnx.numpy_helper) tp (onnx.AttributeProto attribute) train() (onnxruntime.training.ortmodule.ORTModule method) trained_coef_ (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer property) training_info (onnx.ModelProto attribute) training_mode (onnxruntime.capi._pybind_state.RunOptions property) training_optimizer_name (onnxruntime.capi._pybind_state.TrainingParameters property) TrainingAgent (class in onnxruntime.capi._pybind_state) TrainingGraphTransformerConfiguration (class in onnxruntime.capi._pybind_state) TrainingInfoProto (class in onnx) TrainingParameters (class in onnxruntime.capi._pybind_state) TrainingSession (class in onnxruntime) (class in onnxruntime.capi._pybind_state) transfer learning transformer_layer_recompute (onnxruntime.capi._pybind_state.TrainingGraphTransformerConfiguration property) (onnxruntime.capi._pybind_state.TrainingParameters property) tutorial, [1], [2] type (onnx.AttributeProto attribute) (onnx.ValueInfoProto attribute) type() (onnxruntime.training.ortmodule.ORTModule method) type_protos (onnx.AttributeProto attribute) U uint64_data (onnx.TensorProto attribute) unreduced_onnx_loss() (in module onnxcustom.utils.onnx_rewriter) unregister_python_functions() (in module onnxruntime.capi._pybind_state) update_binding (onnx.TrainingInfoProto attribute) update_inplace() (onnxruntime.capi._pybind_state.OrtValue method) (onnxruntime.OrtValue method) update_learning_rate() (onnxcustom.training.sgd_learning_rate.BaseLearningRate method) (onnxcustom.training.sgd_learning_rate.LearningRateSGD method) (onnxcustom.training.sgd_learning_rate.LearningRateSGDNesterov method) update_weights() (onnxcustom.training.sgd_learning_penalty.BaseLearningPenalty method) (onnxcustom.training.sgd_learning_penalty.ElasticLearningPenalty method) (onnxcustom.training.sgd_learning_penalty.NoLearningPenalty method) (onnxcustom.training.sgd_learning_rate.BaseLearningRate method) (onnxcustom.training.sgd_learning_rate.LearningRateSGD method) (onnxcustom.training.sgd_learning_rate.LearningRateSGDNesterov method) use_deterministic_compute (onnxruntime.capi._pybind_state.SessionOptions property) use_fp16_moments (onnxruntime.capi._pybind_state.TrainingParameters property) use_memory_efficient_gradient (onnxruntime.capi._pybind_state.OrtModuleGraphBuilderConfiguration property) (onnxruntime.capi._pybind_state.TrainingParameters property) use_mixed_precision (onnxruntime.capi._pybind_state.TrainingParameters property) V value (onnxcustom.training.sgd_learning_rate.BaseLearningRate property) (onnxcustom.training.sgd_learning_rate.LearningRateSGD property) value_info (onnx.GraphProto attribute) ValueInfoProto (class in onnx) values (onnx.MapProto attribute) (onnx.SparseTensorProto attribute) values() (onnxruntime.capi._pybind_state.SparseTensor method) (onnxruntime.SparseTensor method) version (onnxruntime.capi._pybind_state.ModelMetadata property) W weights_not_to_train (onnxruntime.capi._pybind_state.TrainingParameters property) weights_to_train (onnxruntime.capi._pybind_state.TrainingParameters property) white list WOE, [1] WOEEncoder WOETransformer world_rank (onnxruntime.capi._pybind_state.TrainingParameters property) world_size (onnxruntime.capi._pybind_state.TrainingParameters property) X XGBoost xgboost xpu() (onnxruntime.training.ortmodule.ORTModule method) Z zero_grad() (onnxruntime.training.ortmodule.ORTModule method)