pub struct Split { /* private fields */ }
Expand description
Implementations§
Source§impl Split
impl Split
Sourcepub fn num_split<ArgType: Into<i64>>(self, value: ArgType) -> Self
pub fn num_split<ArgType: Into<i64>>(self, value: ArgType) -> Self
Sets the num_split
attribute.
Sourcepub fn target_device_name<ArgType: Into<String>>(self, value: ArgType) -> Self
pub fn target_device_name<ArgType: Into<String>>(self, value: ArgType) -> Self
Sets the `` attribute.
Sourcepub fn call<'a, T0: ToTensorHandle<'a>, T1: ToTensorHandle<'a>>(
&self,
ctx: &'a Context,
split_dim: &T0,
value: &T1,
) -> Result<TensorHandle<'a>>
pub fn call<'a, T0: ToTensorHandle<'a>, T1: ToTensorHandle<'a>>( &self, ctx: &'a Context, split_dim: &T0, value: &T1, ) -> Result<TensorHandle<'a>>
Execute split.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for Split
impl RefUnwindSafe for Split
impl Send for Split
impl Sync for Split
impl Unpin for Split
impl UnwindSafe for Split
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