pub struct GradientCompressor { /* private fields */ }Implementations§
Source§impl GradientCompressor
impl GradientCompressor
pub fn new(method: CompressionMethod) -> Self
pub fn compress( &mut self, gradients: &HashMap<String, Tensor>, ) -> Result<HashMap<String, CompressedGradient>>
pub fn decompress( &self, compressed: &HashMap<String, CompressedGradient>, ) -> Result<HashMap<String, Tensor>>
pub fn get_compression_ratio(&self) -> f32
pub fn reset_error_buffer(&mut self)
Trait Implementations§
Auto Trait Implementations§
impl Freeze for GradientCompressor
impl RefUnwindSafe for GradientCompressor
impl Send for GradientCompressor
impl Sync for GradientCompressor
impl Unpin for GradientCompressor
impl UnwindSafe for GradientCompressor
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
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
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