Training utilities ================== .. contents:: :local: ONNX ++++ .. autosignature:: onnxcustom.utils.onnx_helper.add_initializer .. autosignature:: onnxcustom.utils.onnx_helper.dtype_to_var_type .. autosignature:: onnxcustom.utils.onnx_helper.get_onnx_opset .. autosignature:: onnxcustom.utils.orttraining_helper.get_train_initializer .. autosignature:: onnxcustom.utils.onnx_helper.proto_type_to_dtype .. autosignature:: onnxcustom.utils.onnx_helper.onnx_rename_weights .. autosignature:: onnxcustom.utils.onnx_rewriter.onnx_rewrite_operator .. autosignature:: onnxcustom.utils.onnx_helper.replace_initializers_into_onnx onnxruntime +++++++++++ .. autosignature:: onnxcustom.utils.onnxruntime_helper.device_to_providers .. autosignature:: onnxcustom.utils.onnxruntime_helper.numpy_to_ort_value .. autosignature:: onnxcustom.utils.onnxruntime_helper.get_ort_device .. autosignature:: onnxcustom.utils.onnxruntime_helper.get_ort_device_type .. autosignature:: onnxcustom.utils.onnxruntime_helper.ort_device_to_string .. autosignature:: onnxcustom.utils.onnxruntime_helper.provider_to_device functions +++++++++ .. autosignature:: onnxcustom.utils.orttraining_helper.add_loss_output .. autosignature:: onnxcustom.utils.onnx_function.get_supported_functions .. autosignature:: onnxcustom.utils.onnx_function.function_onnx_graph .. autosignature:: onnxcustom.utils.orttraining_helper.penalty_loss_onnx gradient ++++++++ .. autosignature:: onnxcustom.training.grad_helper.onnx_derivative