Struct burn_jit::PrecisionBridge

source · 
pub struct PrecisionBridge<R> { /* private fields */ }
Expand description

Handle precision conversion for the jit backend.

Trait Implementations§

source§

impl<ROrigin, RTarget> BackendBridge<JitBackend<ROrigin>> for PrecisionBridge<RTarget>
where ROrigin: Runtime, RTarget: Runtime<Device = ROrigin::Device, Server = ROrigin::Server, Channel = ROrigin::Channel>,

§

type Target = JitBackend<RTarget>

The target backend
source§

fn into_target<const D: usize>( tensor: FloatTensor<JitBackend<ROrigin>, D>, device: Option<Device<Self::Target>> ) -> FloatTensor<Self::Target, D>

Transfer the tensor to the target backend.
source§

fn from_target<const D: usize>( tensor: FloatTensor<Self::Target, D>, device: Option<Device<JitBackend<ROrigin>>> ) -> FloatTensor<JitBackend<ROrigin>, D>

Transfer the tensor from the target backend.
source§

impl<R: Debug> Debug for PrecisionBridge<R>

source§

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

Formats the value using the given formatter. Read more

Auto Trait Implementations§

§

impl<R> Freeze for PrecisionBridge<R>

§

impl<R> RefUnwindSafe for PrecisionBridge<R>
where R: RefUnwindSafe,

§

impl<R> Send for PrecisionBridge<R>
where R: Send,

§

impl<R> Sync for PrecisionBridge<R>
where R: Sync,

§

impl<R> Unpin for PrecisionBridge<R>
where R: Unpin,

§

impl<R> UnwindSafe for PrecisionBridge<R>
where R: UnwindSafe,

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> 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, U> TryFrom<U> for T
where U: Into<T>,

§

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>,

§

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.
source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

source§

fn vzip(self) -> V