pub struct BinaryArgs<'a, T: Element, const N: usize> {
pub a: TensorRef<'a, T, N>,
pub b: TensorRef<'a, T, N>,
pub y: TensorMut<'a, T, N>,
}Expand description
Args bundle for a binary elementwise launch.
Lifetime 'a and rank N are shared across all three tensors; the
element type T is shared too (heterogeneous-dtype ops like
compare(f32, f32) -> bool use a future BinaryWithOutDtypePlan,
not this one).
Fields§
§a: TensorRef<'a, T, N>First input.
b: TensorRef<'a, T, N>Second input.
y: TensorMut<'a, T, N>Output. Aliasing with either input is allowed (in-place add).
Auto Trait Implementations§
impl<'a, T, const N: usize> !UnwindSafe for BinaryArgs<'a, T, N>
impl<'a, T, const N: usize> Freeze for BinaryArgs<'a, T, N>
impl<'a, T, const N: usize> RefUnwindSafe for BinaryArgs<'a, T, N>where
T: RefUnwindSafe,
impl<'a, T, const N: usize> Send for BinaryArgs<'a, T, N>
impl<'a, T, const N: usize> Sync for BinaryArgs<'a, T, N>where
T: Sync,
impl<'a, T, const N: usize> Unpin for BinaryArgs<'a, T, N>
impl<'a, T, const N: usize> UnsafeUnpin for BinaryArgs<'a, T, N>
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