pub enum JaxDeviceType {
Cpu,
Tpu,
Gpu,
}Expand description
Known device types as reported by JAX DLPack tensors.
Variants§
Cpu
Standard host CPU (DLPack device type 1).
Tpu
TPU accelerator (JAX extension, code 13).
Gpu
CUDA GPU (device type 2).
Trait Implementations§
Source§impl Clone for JaxDeviceType
impl Clone for JaxDeviceType
Source§fn clone(&self) -> JaxDeviceType
fn clone(&self) -> JaxDeviceType
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for JaxDeviceType
impl Debug for JaxDeviceType
Source§impl PartialEq for JaxDeviceType
impl PartialEq for JaxDeviceType
impl Copy for JaxDeviceType
impl Eq for JaxDeviceType
impl StructuralPartialEq for JaxDeviceType
Auto Trait Implementations§
impl Freeze for JaxDeviceType
impl RefUnwindSafe for JaxDeviceType
impl Send for JaxDeviceType
impl Sync for JaxDeviceType
impl Unpin for JaxDeviceType
impl UnsafeUnpin for JaxDeviceType
impl UnwindSafe for JaxDeviceType
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> 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 moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.