ONNX Runtime
Loading...
Searching...
No Matches
OrtTrainingApi Member List

This is the complete list of members for OrtTrainingApi, including all inherited members.

AddProperty(OrtCheckpointState *checkpoint_state, const char *property_name, enum OrtPropertyType property_type, void *property_value)OrtTrainingApi
CopyBufferToParameters(OrtTrainingSession *sess, OrtValue *parameters_buffer, bool trainable_only)OrtTrainingApi
CopyParametersToBuffer(OrtTrainingSession *sess, OrtValue *parameters_buffer, bool trainable_only)OrtTrainingApi
CreateTrainingSession(const OrtEnv *env, const OrtSessionOptions *options, OrtCheckpointState *checkpoint_state, const char *train_model_path, const char *eval_model_path, const char *optimizer_model_path, OrtTrainingSession **out)OrtTrainingApi
CreateTrainingSessionFromBuffer(const OrtEnv *env, const OrtSessionOptions *options, OrtCheckpointState *checkpoint_state, const void *train_model_data, size_t train_data_length, const void *eval_model_data, size_t eval_data_length, const void *optim_model_data, size_t optim_data_length, OrtTrainingSession **out)OrtTrainingApi
EvalStep(const OrtTrainingSession *sess, const OrtRunOptions *run_options, size_t inputs_len, const OrtValue *const *inputs, size_t outputs_len, OrtValue **outputs)OrtTrainingApi
ExportModelForInferencing(OrtTrainingSession *sess, const char *inference_model_path, size_t graph_outputs_len, const char *const *graph_output_names)OrtTrainingApi
GetLearningRate(OrtTrainingSession *sess, float *learning_rate)OrtTrainingApi
GetParameter(const OrtCheckpointState *checkpoint_state, const char *parameter_name, OrtAllocator *allocator, OrtValue **parameter)OrtTrainingApi
GetParametersSize(OrtTrainingSession *sess, size_t *out, bool trainable_only)OrtTrainingApi
GetParameterTypeAndShape(const OrtCheckpointState *checkpoint_state, const char *parameter_name, OrtTensorTypeAndShapeInfo **parameter_type_and_shape)OrtTrainingApi
GetProperty(const OrtCheckpointState *checkpoint_state, const char *property_name, OrtAllocator *allocator, enum OrtPropertyType *property_type, void **property_value)OrtTrainingApi
LazyResetGrad(OrtTrainingSession *session)OrtTrainingApi
LoadCheckpoint(const char *checkpoint_path, OrtCheckpointState **checkpoint_state)OrtTrainingApi
LoadCheckpointFromBuffer(const void *checkpoint_buffer, const size_t num_bytes, OrtCheckpointState **checkpoint_state)OrtTrainingApi
OptimizerStep(OrtTrainingSession *sess, const OrtRunOptions *run_options)OrtTrainingApi
RegisterLinearLRScheduler(OrtTrainingSession *sess, const int64_t warmup_step_count, const int64_t total_step_count, const float initial_lr)OrtTrainingApi
ReleaseCheckpointState(OrtCheckpointState *input)OrtTrainingApi
ReleaseTrainingSession(OrtTrainingSession *input)OrtTrainingApi
SaveCheckpoint(OrtCheckpointState *checkpoint_state, const char *checkpoint_path, const bool include_optimizer_state)OrtTrainingApi
SchedulerStep(OrtTrainingSession *sess)OrtTrainingApi
SetLearningRate(OrtTrainingSession *sess, float learning_rate)OrtTrainingApi
SetSeed(const int64_t seed)OrtTrainingApi
TrainingSessionGetEvalModelInputCount(const OrtTrainingSession *sess, size_t *out)OrtTrainingApi
TrainingSessionGetEvalModelInputName(const OrtTrainingSession *sess, size_t index, OrtAllocator *allocator, char **output)OrtTrainingApi
TrainingSessionGetEvalModelOutputCount(const OrtTrainingSession *sess, size_t *out)OrtTrainingApi
TrainingSessionGetEvalModelOutputName(const OrtTrainingSession *sess, size_t index, OrtAllocator *allocator, char **output)OrtTrainingApi
TrainingSessionGetTrainingModelInputCount(const OrtTrainingSession *sess, size_t *out)OrtTrainingApi
TrainingSessionGetTrainingModelInputName(const OrtTrainingSession *sess, size_t index, OrtAllocator *allocator, char **output)OrtTrainingApi
TrainingSessionGetTrainingModelOutputCount(const OrtTrainingSession *sess, size_t *out)OrtTrainingApi
TrainingSessionGetTrainingModelOutputName(const OrtTrainingSession *sess, size_t index, OrtAllocator *allocator, char **output)OrtTrainingApi
TrainStep(OrtTrainingSession *sess, const OrtRunOptions *run_options, size_t inputs_len, const OrtValue *const *inputs, size_t outputs_len, OrtValue **outputs)OrtTrainingApi
UpdateParameter(OrtCheckpointState *checkpoint_state, const char *parameter_name, OrtValue *parameter)OrtTrainingApi