pub struct GraphFreezingConfig {
pub remove_training_ops: bool,
pub fold_constants: bool,
pub remove_dead_code: bool,
pub optimize_sequential_ops: bool,
pub target_device: String,
}Expand description
Configuration for graph freezing optimization
Fields§
§remove_training_ops: boolRemove training-specific operations (dropout, batch norm training mode)
fold_constants: boolFold constants into the graph
remove_dead_code: boolRemove unused nodes
optimize_sequential_ops: boolOptimize sequential operations
target_device: StringTarget device for optimization
Trait Implementations§
Source§impl Clone for GraphFreezingConfig
impl Clone for GraphFreezingConfig
Source§impl Debug for GraphFreezingConfig
impl Debug for GraphFreezingConfig
Auto Trait Implementations§
impl Freeze for GraphFreezingConfig
impl RefUnwindSafe for GraphFreezingConfig
impl Send for GraphFreezingConfig
impl Sync for GraphFreezingConfig
impl Unpin for GraphFreezingConfig
impl UnsafeUnpin for GraphFreezingConfig
impl UnwindSafe for GraphFreezingConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more