OrtTrainingApi

ort_sys

Struct OrtTrainingApi 

Source 
#[repr(C)]
pub struct OrtTrainingApi {
Show 33 fields
    pub LoadCheckpoint: unsafe extern "system" fn(checkpoint_path: *const ortchar, checkpoint_state: *mut *mut OrtCheckpointState) -> OrtStatusPtr,
    pub SaveCheckpoint: unsafe extern "system" fn(checkpoint_state: *mut OrtCheckpointState, checkpoint_path: *const ortchar, include_optimizer_state: bool) -> OrtStatusPtr,
    pub CreateTrainingSession: unsafe extern "system" fn(env: *const OrtEnv, options: *const OrtSessionOptions, checkpoint_state: *mut OrtCheckpointState, train_model_path: *const ortchar, eval_model_path: *const ortchar, optimizer_model_path: *const ortchar, out: *mut *mut OrtTrainingSession) -> OrtStatusPtr,
    pub CreateTrainingSessionFromBuffer: unsafe extern "system" fn(env: *const OrtEnv, options: *const OrtSessionOptions, checkpoint_state: *mut OrtCheckpointState, train_model_data: *const (), train_data_length: usize, eval_model_data: *const (), eval_data_length: usize, optimizer_model_data: *const (), optimizer_data_length: usize, out: *mut *mut OrtTrainingSession) -> OrtStatusPtr,
    pub TrainingSessionGetTrainingModelOutputCount: unsafe extern "system" fn(sess: *const OrtTrainingSession, out: *mut usize) -> OrtStatusPtr,
    pub TrainingSessionGetEvalModelOutputCount: unsafe extern "system" fn(sess: *const OrtTrainingSession, out: *mut usize) -> OrtStatusPtr,
    pub TrainingSessionGetTrainingModelOutputName: unsafe extern "system" fn(sess: *const OrtTrainingSession, index: usize, allocator: *mut OrtAllocator, output: *mut *const c_char) -> OrtStatusPtr,
    pub TrainingSessionGetEvalModelOutputName: unsafe extern "system" fn(sess: *const OrtTrainingSession, index: usize, allocator: *mut OrtAllocator, output: *mut *const c_char) -> OrtStatusPtr,
    pub LazyResetGrad: unsafe extern "system" fn(session: *mut OrtTrainingSession) -> OrtStatusPtr,
    pub TrainStep: unsafe extern "system" fn(session: *mut OrtTrainingSession, run_options: *const OrtRunOptions, inputs_len: usize, inputs: *const *const OrtValue, outputs_len: usize, outputs: *mut *mut OrtValue) -> OrtStatusPtr,
    pub EvalStep: unsafe extern "system" fn(session: *mut OrtTrainingSession, run_options: *const OrtRunOptions, inputs_len: usize, inputs: *const *const OrtValue, outputs_len: usize, outputs: *mut *mut OrtValue) -> OrtStatusPtr,
    pub SetLearningRate: unsafe extern "system" fn(session: *mut OrtTrainingSession, learning_rate: f32) -> OrtStatusPtr,
    pub GetLearningRate: unsafe extern "system" fn(session: *mut OrtTrainingSession, learning_rate: *mut f32) -> OrtStatusPtr,
    pub OptimizerStep: unsafe extern "system" fn(session: *mut OrtTrainingSession, run_options: *const OrtRunOptions) -> OrtStatusPtr,
    pub RegisterLinearLRScheduler: unsafe extern "system" fn(session: *mut OrtTrainingSession, warmup_step_count: i64, total_step_count: i64, initial_lr: f32) -> OrtStatusPtr,
    pub SchedulerStep: unsafe extern "system" fn(session: *mut OrtTrainingSession) -> OrtStatusPtr,
    pub GetParametersSize: unsafe extern "system" fn(session: *mut OrtTrainingSession, out: *mut usize, trainable_only: bool) -> OrtStatusPtr,
    pub CopyParametersToBuffer: unsafe extern "system" fn(session: *mut OrtTrainingSession, parameters_buffer: *mut OrtValue, trainable_only: bool) -> OrtStatusPtr,
    pub CopyBufferToParameters: unsafe extern "system" fn(session: *mut OrtTrainingSession, parameters_buffer: *mut OrtValue, trainable_only: bool) -> OrtStatusPtr,
    pub ReleaseTrainingSession: unsafe extern "system" fn(input: *mut OrtTrainingSession),
    pub ReleaseCheckpointState: unsafe extern "system" fn(input: *mut OrtCheckpointState),
    pub ExportModelForInferencing: unsafe extern "system" fn(session: *mut OrtTrainingSession, inference_model_path: *const ortchar, graph_outputs_len: usize, graph_output_names: *const *const c_char) -> OrtStatusPtr,
    pub SetSeed: unsafe extern "system" fn(seed: i64) -> OrtStatusPtr,
    pub TrainingSessionGetTrainingModelInputCount: unsafe extern "system" fn(session: *const OrtTrainingSession, out: *mut usize) -> OrtStatusPtr,
    pub TrainingSessionGetEvalModelInputCount: unsafe extern "system" fn(session: *const OrtTrainingSession, out: *mut usize) -> OrtStatusPtr,
    pub TrainingSessionGetTrainingModelInputName: unsafe extern "system" fn(session: *const OrtTrainingSession, index: usize, allocator: *mut OrtAllocator, output: *mut *const c_char) -> OrtStatusPtr,
    pub TrainingSessionGetEvalModelInputName: unsafe extern "system" fn(session: *const OrtTrainingSession, index: usize, allocator: *mut OrtAllocator, output: *mut *const c_char) -> OrtStatusPtr,
    pub AddProperty: unsafe extern "system" fn(checkpoint_state: *mut OrtCheckpointState, property_name: *const c_char, property_type: OrtPropertyType, property_value: *const ()) -> OrtStatusPtr,
    pub GetProperty: unsafe extern "system" fn(checkpoint_state: *mut OrtCheckpointState, property_name: *const c_char, allocator: *mut OrtAllocator, property_type: *mut OrtPropertyType, property_value: *mut *const ()) -> OrtStatusPtr,
    pub LoadCheckpointFromBuffer: unsafe extern "system" fn(checkpoint_buffer: *const (), num_bytes: usize, checkpoint_state: *mut *mut OrtCheckpointState) -> OrtStatusPtr,
    pub GetParameterTypeAndShape: unsafe extern "system" fn(checkpoint_state: *const OrtCheckpointState, parameter_name: *const c_char, parameter_type_and_shape: *mut *mut OrtTensorTypeAndShapeInfo) -> OrtStatusPtr,
    pub UpdateParameter: unsafe extern "system" fn(checkpoint_state: *mut OrtCheckpointState, parameter_name: *const c_char, parameter: *mut OrtValue) -> OrtStatusPtr,
    pub GetParameter: unsafe extern "system" fn(checkpoint_state: *const OrtCheckpointState, parameter_name: *const c_char, allocator: *mut OrtAllocator, parameter: *mut *mut OrtValue) -> OrtStatusPtr,

}

Fields§

§LoadCheckpoint: unsafe extern "system" fn(checkpoint_path: *const ortchar, checkpoint_state: *mut *mut OrtCheckpointState) -> OrtStatusPtr§SaveCheckpoint: unsafe extern "system" fn(checkpoint_state: *mut OrtCheckpointState, checkpoint_path: *const ortchar, include_optimizer_state: bool) -> OrtStatusPtr§CreateTrainingSession: unsafe extern "system" fn(env: *const OrtEnv, options: *const OrtSessionOptions, checkpoint_state: *mut OrtCheckpointState, train_model_path: *const ortchar, eval_model_path: *const ortchar, optimizer_model_path: *const ortchar, out: *mut *mut OrtTrainingSession) -> OrtStatusPtr§CreateTrainingSessionFromBuffer: unsafe extern "system" fn(env: *const OrtEnv, options: *const OrtSessionOptions, checkpoint_state: *mut OrtCheckpointState, train_model_data: *const (), train_data_length: usize, eval_model_data: *const (), eval_data_length: usize, optimizer_model_data: *const (), optimizer_data_length: usize, out: *mut *mut OrtTrainingSession) -> OrtStatusPtr§TrainingSessionGetTrainingModelOutputCount: unsafe extern "system" fn(sess: *const OrtTrainingSession, out: *mut usize) -> OrtStatusPtr§TrainingSessionGetEvalModelOutputCount: unsafe extern "system" fn(sess: *const OrtTrainingSession, out: *mut usize) -> OrtStatusPtr§TrainingSessionGetTrainingModelOutputName: unsafe extern "system" fn(sess: *const OrtTrainingSession, index: usize, allocator: *mut OrtAllocator, output: *mut *const c_char) -> OrtStatusPtr§TrainingSessionGetEvalModelOutputName: unsafe extern "system" fn(sess: *const OrtTrainingSession, index: usize, allocator: *mut OrtAllocator, output: *mut *const c_char) -> OrtStatusPtr§LazyResetGrad: unsafe extern "system" fn(session: *mut OrtTrainingSession) -> OrtStatusPtr§TrainStep: unsafe extern "system" fn(session: *mut OrtTrainingSession, run_options: *const OrtRunOptions, inputs_len: usize, inputs: *const *const OrtValue, outputs_len: usize, outputs: *mut *mut OrtValue) -> OrtStatusPtr§EvalStep: unsafe extern "system" fn(session: *mut OrtTrainingSession, run_options: *const OrtRunOptions, inputs_len: usize, inputs: *const *const OrtValue, outputs_len: usize, outputs: *mut *mut OrtValue) -> OrtStatusPtr§SetLearningRate: unsafe extern "system" fn(session: *mut OrtTrainingSession, learning_rate: f32) -> OrtStatusPtr§GetLearningRate: unsafe extern "system" fn(session: *mut OrtTrainingSession, learning_rate: *mut f32) -> OrtStatusPtr§OptimizerStep: unsafe extern "system" fn(session: *mut OrtTrainingSession, run_options: *const OrtRunOptions) -> OrtStatusPtr§RegisterLinearLRScheduler: unsafe extern "system" fn(session: *mut OrtTrainingSession, warmup_step_count: i64, total_step_count: i64, initial_lr: f32) -> OrtStatusPtr§SchedulerStep: unsafe extern "system" fn(session: *mut OrtTrainingSession) -> OrtStatusPtr§GetParametersSize: unsafe extern "system" fn(session: *mut OrtTrainingSession, out: *mut usize, trainable_only: bool) -> OrtStatusPtr§CopyParametersToBuffer: unsafe extern "system" fn(session: *mut OrtTrainingSession, parameters_buffer: *mut OrtValue, trainable_only: bool) -> OrtStatusPtr§CopyBufferToParameters: unsafe extern "system" fn(session: *mut OrtTrainingSession, parameters_buffer: *mut OrtValue, trainable_only: bool) -> OrtStatusPtr§ReleaseTrainingSession: unsafe extern "system" fn(input: *mut OrtTrainingSession)§ReleaseCheckpointState: unsafe extern "system" fn(input: *mut OrtCheckpointState)§ExportModelForInferencing: unsafe extern "system" fn(session: *mut OrtTrainingSession, inference_model_path: *const ortchar, graph_outputs_len: usize, graph_output_names: *const *const c_char) -> OrtStatusPtr§SetSeed: unsafe extern "system" fn(seed: i64) -> OrtStatusPtr§TrainingSessionGetTrainingModelInputCount: unsafe extern "system" fn(session: *const OrtTrainingSession, out: *mut usize) -> OrtStatusPtr§TrainingSessionGetEvalModelInputCount: unsafe extern "system" fn(session: *const OrtTrainingSession, out: *mut usize) -> OrtStatusPtr§TrainingSessionGetTrainingModelInputName: unsafe extern "system" fn(session: *const OrtTrainingSession, index: usize, allocator: *mut OrtAllocator, output: *mut *const c_char) -> OrtStatusPtr§TrainingSessionGetEvalModelInputName: unsafe extern "system" fn(session: *const OrtTrainingSession, index: usize, allocator: *mut OrtAllocator, output: *mut *const c_char) -> OrtStatusPtr§AddProperty: unsafe extern "system" fn(checkpoint_state: *mut OrtCheckpointState, property_name: *const c_char, property_type: OrtPropertyType, property_value: *const ()) -> OrtStatusPtr§GetProperty: unsafe extern "system" fn(checkpoint_state: *mut OrtCheckpointState, property_name: *const c_char, allocator: *mut OrtAllocator, property_type: *mut OrtPropertyType, property_value: *mut *const ()) -> OrtStatusPtr§LoadCheckpointFromBuffer: unsafe extern "system" fn(checkpoint_buffer: *const (), num_bytes: usize, checkpoint_state: *mut *mut OrtCheckpointState) -> OrtStatusPtr§GetParameterTypeAndShape: unsafe extern "system" fn(checkpoint_state: *const OrtCheckpointState, parameter_name: *const c_char, parameter_type_and_shape: *mut *mut OrtTensorTypeAndShapeInfo) -> OrtStatusPtr§UpdateParameter: unsafe extern "system" fn(checkpoint_state: *mut OrtCheckpointState, parameter_name: *const c_char, parameter: *mut OrtValue) -> OrtStatusPtr§GetParameter: unsafe extern "system" fn(checkpoint_state: *const OrtCheckpointState, parameter_name: *const c_char, allocator: *mut OrtAllocator, parameter: *mut *mut OrtValue) -> OrtStatusPtr

Trait Implementations§

Source§

impl Clone for OrtTrainingApi

Source§

fn clone(&self) -> OrtTrainingApi

Returns a duplicate of the value. Read more
1.0.0 · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl Debug for OrtTrainingApi

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more
Source§

impl Copy for OrtTrainingApi

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.