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 2022-12 _ __abstractmethods__ (onnxruntime.quantization.calibrate.CalibrationDataReader attribute) __annotations__ (onnx.backend.base.DeviceType attribute) (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__() (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType 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) __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__() (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType method) (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__() (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType 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) __index__() (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType 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) __init__() (onnx.backend.base.Device method) (onnx.defs.OpSchema method) (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType method) (onnx.hub.ModelInfo method) (onnx.reference.op_run.SparseTensor method) (onnx.utils.Extractor 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) (onnxcustom.utils.onnx_split.OnnxSegment method) (onnxcustom.utils.onnx_split.OnnxSplitting 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__() (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType 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) __iter__() (onnxruntime.capi._pybind_state.OrtValueVector method) (onnxruntime.quantization.calibrate.CalibrationDataReader 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__ (onnx.defs.OpSchema.AttrType attribute) (onnx.defs.OpSchema.DifferentiationCategory attribute) (onnx.defs.OpSchema.FormalParameterOption attribute) (onnx.defs.OpSchema.SupportType attribute) (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__() (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType 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) __new__() (onnxruntime.InferenceSession method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) __next__() (onnxruntime.quantization.calibrate.CalibrationDataReader 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.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType method) (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) (onnxcustom.utils.onnx_split.OnnxSegment 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__() (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType method) (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.OperatorProto attribute) (onnx.OperatorSetIdProto attribute) (onnx.OperatorSetProto attribute) (onnx.OptionalProto attribute) (onnx.SequenceProto attribute) (onnx.SparseTensorProto attribute) (onnx.StringStringEntryProto attribute) (onnx.TensorProto attribute) (onnx.TensorProto.Segment attribute) (onnx.TensorShapeProto attribute) (onnx.TensorShapeProto.Dimension attribute) (onnx.TrainingInfoProto attribute) (onnx.TypeProto attribute) (onnx.TypeProto.Map attribute) (onnx.TypeProto.Opaque attribute) (onnx.TypeProto.Optional attribute) (onnx.TypeProto.Sequence attribute) (onnx.TypeProto.SparseTensor attribute) (onnx.TypeProto.Tensor attribute) (onnx.ValueInfoProto attribute) __str__() (onnx.defs.OpSchema.AttrType method) (onnx.defs.OpSchema.DifferentiationCategory method) (onnx.defs.OpSchema.FormalParameterOption method) (onnx.defs.OpSchema.SupportType method) (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.quantization.calibrate.CalibrationDataReader class method) (onnxruntime.training.ortmodule.ORTModule method) (onnxruntime.TrainingSession method) _abc_impl (onnxruntime.quantization.calibrate.CalibrationDataReader attribute) _apply() (onnxruntime.training.ortmodule.ORTModule method) _backward_hooks (onnxruntime.training.ortmodule.ORTModule attribute) _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) _buffers (onnxruntime.training.ortmodule.ORTModule attribute) _build_name2obj_dict() (onnx.utils.Extractor 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) _collect_new_inputs() (onnx.utils.Extractor method) _collect_new_io_core() (onnx.utils.Extractor method) _collect_new_outputs() (onnx.utils.Extractor method) _collect_reachable_nodes() (onnx.utils.Extractor method) _collect_reachable_tensors() (onnx.utils.Extractor method) _collect_referred_local_functions() (onnx.utils.Extractor method) _connex_components() (onnxcustom.utils.onnx_split.OnnxSplitting static 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) _dfs_search_reachable_nodes() (onnx.utils.Extractor method) _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) _forward_hooks (onnxruntime.training.ortmodule.ORTModule attribute) _forward_pre_hooks (onnxruntime.training.ortmodule.ORTModule attribute) _function_body (onnx.defs.OpSchema property) _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_cutting_points() (onnxcustom.utils.onnx_split.OnnxSplitting 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) _infer_node_outputs() (onnx.defs.OpSchema method) _init() (onnxcustom.utils.onnx_split.OnnxSplitting 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) _key() (onnxcustom.utils.onnx_split.OnnxSplitting static method) _load_from_state_dict() (onnxruntime.training.ortmodule.ORTModule method) _load_state_dict_post_hooks (onnxruntime.training.ortmodule.ORTModule attribute) _load_state_dict_pre_hooks (onnxruntime.training.ortmodule.ORTModule attribute) _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) _make_model() (onnx.utils.Extractor method) _make_onnx() (onnxcustom.utils.onnx_split.OnnxSplitting method) _make_segment() (onnxcustom.utils.onnx_split.OnnxSplitting method) _maybe_warn_non_full_backward_hook() (onnxruntime.training.ortmodule.ORTModule method) _modules (onnxruntime.training.ortmodule.ORTModule attribute) _munge_time() (in module onnxcustom.utils.nvprof2json) _named_members() (onnxruntime.training.ortmodule.ORTModule method) _next_iter() (onnxcustom.training.data_loader.OrtDataLoader method) _non_persistent_buffers_set (onnxruntime.training.ortmodule.ORTModule attribute) _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) _parameters (onnxruntime.training.ortmodule.ORTModule attribute) _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) _split_2() (onnxcustom.utils.onnx_split.OnnxSplitting method) _state_dict_hooks (onnxruntime.training.ortmodule.ORTModule attribute) _Type (onnx.backend.base.DeviceType attribute) _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 Backend (class in onnx.backend.base) BackendRep (class in onnx.backend.base) 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) big models 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 CalibrationDataReader (class in onnxruntime.quantization.calibrate) 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_model_from_external_data() (in module onnx.external_data_helper) convert_model_to_external_data() (in module onnx.external_data_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() (onnx.reference.op_run.OpFunction class method) (onnx.reference.op_run.OpRun class 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) DefaultNone (class in onnx.reference.op_run) denotation (onnx.TensorShapeProto.Dimension attribute) (onnx.TypeProto attribute) 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.OperatorProto attribute) (onnx.OperatorSetIdProto attribute) (onnx.OperatorSetProto attribute) (onnx.OptionalProto attribute) (onnx.SequenceProto attribute) (onnx.SparseTensorProto attribute) (onnx.StringStringEntryProto attribute) (onnx.TensorProto attribute) (onnx.TensorProto.Segment attribute) (onnx.TensorShapeProto attribute) (onnx.TensorShapeProto.Dimension attribute) (onnx.TrainingInfoProto attribute) (onnx.TypeProto attribute) (onnx.TypeProto.Map attribute) (onnx.TypeProto.Opaque attribute) (onnx.TypeProto.Optional attribute) (onnx.TypeProto.Sequence attribute) (onnx.TypeProto.SparseTensor attribute) (onnx.TypeProto.Tensor attribute) (onnx.ValueInfoProto attribute) Device (class in onnx.backend.base) 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) DeviceType (class in onnx.backend.base) dim (onnx.TensorShapeProto attribute) dim_param (onnx.TensorShapeProto.Dimension attribute) dim_value (onnx.TensorShapeProto.Dimension attribute) 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.OperatorProto attribute) (onnx.OperatorSetProto attribute) (onnx.TensorProto attribute) (onnx.ValueInfoProto attribute) domain (onnx.FunctionProto attribute) (onnx.ModelProto attribute) (onnx.NodeProto attribute) (onnx.OperatorSetIdProto attribute) (onnx.OperatorSetProto attribute) (onnx.reference.op_run.OpFunction property) (onnx.reference.op_run.OpRun property) (onnx.TypeProto.Opaque 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) E ElasticLearningLoss (class in onnxcustom.training.sgd_learning_loss) ElasticLearningPenalty (class in onnxcustom.training.sgd_learning_penalty) elem_type (onnx.OptionalProto attribute) (onnx.SequenceProto attribute) (onnx.TypeProto.Optional attribute) (onnx.TypeProto.Sequence attribute) (onnx.TypeProto.SparseTensor attribute) (onnx.TypeProto.Tensor 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() (onnx.reference.op_run.OpFunction class method) (onnx.reference.op_run.OpRun class method) (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) ExternalDataInfo (class in onnx.external_data_helper) extra_repr() (onnxruntime.training.ortmodule.ORTModule method) extract_model() (in module onnx.utils) Extractor (class in 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) (onnx.OperatorSetProto 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_with_history() (in module onnx.defs) get_all_tensor_dtypes() (in module onnx.helper) get_attribute_value() (in module onnx.helper), [1] 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_context_dependent_function() (onnx.defs.OpSchema method) get_context_dependent_function_with_opset_version() (onnx.defs.OpSchema method) 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_forward_model() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method) get_full_state() (onnxcustom.training.optimizers_partial.OrtGradientForwardBackwardOptimizer method) get_function_ops() (in module onnx.defs) get_function_with_opset_version() (onnx.defs.OpSchema method) get_gradient_model() (onnxruntime.capi._pybind_state.OrtModuleGraphBuilder method) get_graph_info() (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) 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_next() (onnxruntime.quantization.calibrate.CalibrationDataReader 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_from_session() (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) 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) gpu 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_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) implicit_inputs() (onnx.reference.op_run.OpFunction static method) (onnx.reference.op_run.OpRun static method) 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) (onnx.reference.op_run.OpFunction property) (onnx.reference.op_run.OpRun property) input_names (onnx.reference.ReferenceEvaluator property) 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) ipu() (onnxruntime.training.ortmodule.ORTModule method) ir_build_metadata (onnx.OperatorSetProto attribute) ir_version (onnx.ModelProto attribute) (onnx.OperatorSetProto attribute) ir_version_prerelease (onnx.OperatorSetProto attribute) is_dlpack_uint8_tensor() (in module onnxruntime.capi._pybind_state) is_infinite() (onnx.defs.OpSchema static method) is_output_fp32_node() (onnxruntime.capi._pybind_state.TrainingSession method) (onnxruntime.TrainingSession method) is_small() (onnxcustom.utils.onnx_split.OnnxSplitting static 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 (onnx.StringStringEntryProto attribute) key_type (onnx.MapProto attribute) (onnx.TypeProto.Map 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], [2], [3] lightgbm list_models() (in module onnx.hub) load() (in module onnx) (in module onnx.hub) load_external_data_for_model() (in module onnx.external_data_helper) load_external_data_for_tensor() (in module onnx.external_data_helper) load_model() (onnxruntime.capi._pybind_state.TrainingSession method) load_model_from_string() (in module onnx) load_model_tests() (in module onnx.backend.test.loader) 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 magic (onnx.OperatorSetProto attribute) 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) (onnx.reference.op_run.OpFunction class method) (onnx.reference.op_run.OpRun class method) make_onnx() (onnxcustom.utils.onnx_split.OnnxSplitting method) 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_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_type (onnx.TypeProto attribute) map_value (onnx.OptionalProto attribute) 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], [2] module 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.onnx_split 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.defs.OpSchema.AttrType property) (onnx.defs.OpSchema.DifferentiationCategory property) (onnx.defs.OpSchema.FormalParameterOption property) (onnx.defs.OpSchema.SupportType property) (onnx.FunctionProto attribute) (onnx.GraphProto attribute) (onnx.MapProto attribute) (onnx.NodeProto attribute) (onnx.OptionalProto attribute) (onnx.SequenceProto attribute) (onnx.TensorProto attribute) (onnx.TypeProto.Opaque 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) need_context() (onnx.reference.op_run.OpFunction method) (onnx.reference.op_run.OpRun 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) np_dtype_to_tensor_dtype() (in module onnx.helper) 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_derivative() (in module onnxcustom.training.grad_helper) onnx_opset_version() (in module onnx.defs) 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.onnx_split module onnxcustom.utils.onnxruntime_helper module onnxcustom.utils.orttraining_helper module onnxcustom.utils.print_helper module onnxruntime, [1], [2] OnnxSegment (class in onnxcustom.utils.onnx_split) OnnxSplitting (class in onnxcustom.utils.onnx_split) op_type (onnx.NodeProto attribute) (onnx.OperatorProto attribute) opaque_type (onnx.TypeProto attribute) operator (onnx.OperatorSetProto attribute) OperatorProto (class in onnx) OperatorSetIdProto (class in onnx) OperatorSetProto (class in onnx) OpFunction (class in onnx.reference.op_run) OpRun (class in onnx.reference.op_run) OpSchema (class in onnx.defs) OpSchema.AttrType (class in onnx.defs) OpSchema.DifferentiationCategory (class in onnx.defs) OpSchema.FormalParameterOption (class in onnx.defs) OpSchema.SupportType (class in onnx.defs) opset (onnx.hub.ModelInfo attribute) opset_import (onnx.FunctionProto attribute) (onnx.ModelProto attribute) opset_version (onnx.OperatorSetProto attribute) opsets (onnx.reference.ReferenceEvaluator property) 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_type (onnx.TypeProto attribute) optional_value (onnx.OptionalProto attribute) optional_values (onnx.SequenceProto attribute) OptionalProto (class in onnx) 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) (onnx.reference.op_run.OpFunction property) (onnx.reference.op_run.OpRun property) output_names (onnx.reference.ReferenceEvaluator property) outputs_meta (onnxruntime.capi._pybind_state.InferenceSession property) overridable_initializers (onnxruntime.capi._pybind_state.InferenceSession property) P parallel, [1] parameters() (onnxruntime.training.ortmodule.ORTModule method) parse_function() (in module onnx.parser) parse_graph() (in module onnx.parser) parse_model() (in module onnx.parser) ParseFromString() (onnx.ModelProto 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), [1] printable_dim() (in module onnx.helper), [1] printable_graph() (in module onnx.helper), [1] printable_node() (in module onnx.helper), [1] printable_tensor_proto() (in module onnx.helper), [1] printable_type() (in module onnx.helper), [1] printable_value_info() (in module onnx.helper), [1] 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) push_back_batch() (onnxruntime.capi._pybind_state.OrtValueVector method) pytorch_export_helpers() (in module onnxruntime.tools) Q quant_pre_process() (in module onnxruntime.quantization.shape_inference) QuantFormat (class in onnxruntime.quantization.quant_utils) quantization quantization_annotation (onnx.GraphProto attribute) QuantizationMode (class in onnxruntime.quantization.quant_utils) quantize_dynamic() (in module onnxruntime.quantization.quantize) quantize_static() (in module onnxruntime.quantization.quantize) QuantType (class in onnxruntime.quantization.quant_utils) R raw_data (onnx.TensorProto attribute) read_bytes() (onnxruntime.capi._pybind_state.TrainingSession method) ref_attr_name (onnx.AttributeProto attribute) ReferenceEvaluator (class in onnx.reference) 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_load_state_dict_post_hook() (onnxruntime.training.ortmodule.ORTModule method) register_module() (onnxruntime.training.ortmodule.ORTModule method) register_parameter() (onnxruntime.training.ortmodule.ORTModule method) remove() (onnxruntime.capi._pybind_state.OrtValueCache method) remove_external_data_field() (in module onnx.external_data_helper) 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() (onnx.backend.base.BackendRep method) (onnx.reference.op_run.OpFunction method) (onnx.reference.op_run.OpRun method) (onnx.reference.ReferenceEvaluator method) (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_node() (onnx.backend.base.Backend class 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) run_with_ortvaluevector() (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) RuntimeTypeError (class in onnx.reference.op_run) S s (onnx.AttributeProto attribute) save() (onnxruntime.capi._pybind_state.GradientGraphBuilder method) save_external_data() (in module onnx.external_data_helper) 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) SchemaError (class in onnx.defs) 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_type (onnx.TypeProto attribute) sequence_value (onnx.OptionalProto attribute) sequence_values (onnx.SequenceProto attribute) SequenceProto (class in onnx) SerializeToString() (onnx.ModelProto 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_external_data() (in module onnx.external_data_helper) 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 (onnx.TypeProto.SparseTensor attribute) (onnx.TypeProto.Tensor attribute) 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 since_version (onnx.OperatorProto attribute) 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_type (onnx.TypeProto attribute) sparse_tensor_value (onnx.OptionalProto 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 onnx.reference.op_run) (class in onnxruntime) (class in onnxruntime.capi._pybind_state) SparseTensorProto (class in onnx) split_complex_to_pairs() (in module onnx.helper) split_onnx() (in module onnxcustom.utils.onnx_split) split_segment() (onnxcustom.utils.onnx_split.OnnxSplitting method) SquareLearningLoss (class in onnxcustom.training.sgd_learning_loss) state_dict() (onnxruntime.training.ortmodule.ORTModule method) status (onnx.OperatorProto attribute) str_ortvalue() (in module onnxcustom.utils.print_helper) string_data (onnx.TensorProto attribute) string_keys (onnx.MapProto attribute) strings (onnx.AttributeProto attribute) StringStringEntryProto (class in onnx) supports_device() (onnx.backend.base.Backend class method) synchronize_execution_providers (onnxruntime.capi._pybind_state.RunOptions property) 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 td-idf tensor_dtype_to_field() (in module onnx.helper) tensor_dtype_to_np_dtype() (in module onnx.helper) tensor_dtype_to_storage_tensor_dtype() (in module onnx.helper) tensor_dtype_to_string() (in module onnx.helper) tensor_type (onnx.TypeProto attribute) tensor_value (onnx.OptionalProto attribute) tensor_values (onnx.SequenceProto attribute) TensorDtypeMap (class in onnx.mapping) TensorProto (class in onnx) TensorProto.Segment (class in onnx) tensors (onnx.AttributeProto attribute) TensorShapeProto (class in onnx) TensorShapeProto.Dimension (class in onnx) terminate (onnxruntime.capi._pybind_state.RunOptions property) thread, [1] 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) to_text() (in module onnx.printer) 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], [3] type (onnx.AttributeProto attribute) (onnx.ValueInfoProto attribute) type() (onnxruntime.training.ortmodule.ORTModule method) type_protos (onnx.AttributeProto attribute) TypeProto (class in onnx) TypeProto.Map (class in onnx) TypeProto.Opaque (class in onnx) TypeProto.Optional (class in onnx) TypeProto.Sequence (class in onnx) TypeProto.SparseTensor (class in onnx) TypeProto.Tensor (class in onnx) 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_inputs_outputs_dims() (in module onnx.tools.update_model_dims) 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 (onnx.StringStringEntryProto attribute) (onnxcustom.training.sgd_learning_rate.BaseLearningRate property) (onnxcustom.training.sgd_learning_rate.LearningRateSGD property) value_info (onnx.GraphProto attribute) value_type (onnx.TypeProto.Map attribute) ValueInfoProto (class in onnx) values (onnx.MapProto attribute) (onnx.SparseTensorProto attribute) values() (onnxruntime.capi._pybind_state.SparseTensor method) (onnxruntime.SparseTensor method) version (onnx.OperatorSetIdProto attribute) (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) write_external_data_tensors() (in module onnx.external_data_helper) X XGBoost xgboost xpu() (onnxruntime.training.ortmodule.ORTModule method) Z zero_grad() (onnxruntime.training.ortmodule.ORTModule method)