Struct kn_cuda_sys::wrapper::group::TensorOpArgs
source · pub struct TensorOpArgs<P = DevicePtr> {
pub op_desc: TensorOpDescriptor,
pub alpha_1: f32,
pub input_1_desc: TensorDescriptor,
pub input_1_ptr: P,
pub alpha_2: f32,
pub input_2_desc: TensorDescriptor,
pub input_2_ptr: P,
pub beta: f32,
pub output_desc: TensorDescriptor,
pub output_ptr: P,
}Fields§
§op_desc: TensorOpDescriptor§alpha_1: f32§input_1_desc: TensorDescriptor§input_1_ptr: P§alpha_2: f32§input_2_desc: TensorDescriptor§input_2_ptr: P§beta: f32§output_desc: TensorDescriptor§output_ptr: PImplementations§
source§impl TensorOpArgs
impl TensorOpArgs
pub unsafe fn run(&self, handle: &CudnnHandle)
Trait Implementations§
Auto Trait Implementations§
impl<P> RefUnwindSafe for TensorOpArgs<P>where
P: RefUnwindSafe,
impl<P = DevicePtr> !Send for TensorOpArgs<P>
impl<P = DevicePtr> !Sync for TensorOpArgs<P>
impl<P> Unpin for TensorOpArgs<P>where
P: Unpin,
impl<P> UnwindSafe for TensorOpArgs<P>where
P: UnwindSafe,
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