ONNX Runtime
|
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 |