pub struct TestInput { /* private fields */ }Implementations§
Source§impl TestInput
impl TestInput
pub fn random( client: ComputeClient<TestRuntime>, shape: Vec<usize>, dtype: StorageType, seed: u64, distribution: Distribution, stride_spec: StrideSpec, ) -> Self
pub fn zeros( client: ComputeClient<TestRuntime>, shape: Vec<usize>, dtype: StorageType, stride_spec: StrideSpec, ) -> Self
pub fn eye( client: ComputeClient<TestRuntime>, shape: Vec<usize>, dtype: StorageType, stride_spec: StrideSpec, ) -> Self
pub fn arange( client: ComputeClient<TestRuntime>, shape: Vec<usize>, dtype: StorageType, stride_spec: StrideSpec, ) -> Self
pub fn custom( client: ComputeClient<TestRuntime>, shape: Vec<usize>, dtype: StorageType, stride_spec: StrideSpec, data: Vec<f32>, ) -> Self
pub fn generate_with_f32_host_data( self, ) -> (TensorHandle<TestRuntime>, HostData)
pub fn generate_with_bool_host_data( self, ) -> (TensorHandle<TestRuntime>, HostData)
pub fn f32_host_data(self) -> HostData
pub fn bool_host_data(self) -> HostData
pub fn generate_without_host_data(self) -> TensorHandle<TestRuntime>
pub fn generate(self) -> TensorHandle<TestRuntime>
Auto Trait Implementations§
impl Freeze for TestInput
impl !RefUnwindSafe for TestInput
impl Send for TestInput
impl Sync for TestInput
impl Unpin for TestInput
impl !UnwindSafe for TestInput
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