Skip to main content

BridgeTensor

Struct BridgeTensor 

pub struct BridgeTensor { /* private fields */ }
Expand description

A type-tagged tensor at the bridge layer between the high-level tensor API and the dispatch system.

BridgeTensor serves as the runtime representation for the public tensor kinds (Float, Int, Bool) and internal variants like quantized floats, wrapping the uniform DispatchTensor used by the underlying dispatch layer. This separation keeps tensor kind tracking out of the backends while avoiding exposure of backend-level primitives in the public API.

Implementations§

§

impl BridgeTensor

pub fn float(tensor: DispatchTensor) -> BridgeTensor

Builds a bridge tensor that wraps a floating-point dispatch tensor.

Available with the extension feature for backend-extension authors.

pub fn int(tensor: DispatchTensor) -> BridgeTensor

Builds a bridge tensor that wraps an integer dispatch tensor.

Available with the extension feature for backend-extension authors.

pub fn bool(tensor: DispatchTensor) -> BridgeTensor

Builds a bridge tensor that wraps a boolean dispatch tensor.

Available with the extension feature for backend-extension authors.

pub fn qfloat(tensor: DispatchTensor) -> BridgeTensor

Builds a bridge tensor that wraps a quantized floating-point dispatch tensor.

Available with the extension feature for backend-extension authors.

pub fn kind(&self) -> BridgeKind

Returns the runtime tag identifying which variant this tensor wraps.

pub fn is_float(&self) -> bool

Returns true if this tensor is the float variant.

pub fn is_int(&self) -> bool

Returns true if this tensor is the int variant.

pub fn is_bool(&self) -> bool

Returns true if this tensor is the bool variant.

pub fn is_qfloat(&self) -> bool

Returns true if this tensor is the quantized float variant.

pub fn into_parts(self) -> (BridgeKind, DispatchTensor)

Consumes the bridge tensor and returns its variant tag together with the underlying dispatch tensor.

Available with the extension feature for backend-extension authors.

pub fn as_parts(&self) -> (BridgeKind, &DispatchTensor)

Borrows the bridge tensor as its variant tag together with a reference to the underlying dispatch tensor.

Available with the extension feature for backend-extension authors.

pub fn dtype(&self) -> DType

Returns the dtype of the tensor.

pub fn can_mut(&self) -> bool

Whether the tensor’s buffer can be mutated in place (see TensorMetadata::can_mut).

pub fn shape(&self) -> Shape

Returns the shape of the tensor.

pub fn rank(&self) -> usize

Returns the number of dimensions of the tensor.

Trait Implementations§

§

impl Clone for BridgeTensor

§

fn clone(&self) -> BridgeTensor

Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§

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

Performs copy-assignment from source. Read more
§

impl Debug for BridgeTensor

§

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

Formats the value using the given formatter. Read more
§

impl From<BridgeTensor> for DispatchTensor

§

fn from(value: BridgeTensor) -> DispatchTensor

Converts to this type from the input type.

Auto Trait Implementations§

Blanket Implementations§

§

impl<T> Adaptor<()> for T

§

fn adapt(&self)

Adapt the type to be passed to a metric.
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
§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

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
§

impl<K, Q> Comparable<Q> for K
where K: Borrow<Q> + ?Sized, Q: Ord + ?Sized,

§

fn compare(&self, key: &Q) -> Ordering

Compare self to key and return their ordering.
§

impl<K, Q> Equivalent<Q> for K
where K: Borrow<Q> + ?Sized, Q: Eq + ?Sized,

§

fn equivalent(&self, key: &Q) -> bool

Compare self to key and return true if they are equal.
§

impl<T> ErasedDestructor for T
where T: 'static,

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

§

impl<T> Instrument for T

§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided [Span], returning an Instrumented wrapper. Read more
§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
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> IntoEither for T

Source§

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 more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

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
§

impl<T> Pointable for T

§

const ALIGN: usize

The alignment of pointer.
§

type Init = T

The type for initializers.
§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
§

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

§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns [Action::Follow] only if self and other return Action::Follow. Read more
§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns [Action::Follow] if either self or other returns Action::Follow. Read more
§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
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.
§

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

§

fn vzip(self) -> V

§

impl<T> WithSubscriber for T

§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a [WithDispatch] wrapper. Read more
§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a [WithDispatch] wrapper. Read more