pub struct HipTensorLinalgBackend;Expand description
Trait Implementations§
Source§impl Clone for HipTensorLinalgBackend
impl Clone for HipTensorLinalgBackend
Source§fn clone(&self) -> HipTensorLinalgBackend
fn clone(&self) -> HipTensorLinalgBackend
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for HipTensorLinalgBackend
impl Debug for HipTensorLinalgBackend
Source§impl Default for HipTensorLinalgBackend
impl Default for HipTensorLinalgBackend
Source§fn default() -> HipTensorLinalgBackend
fn default() -> HipTensorLinalgBackend
Returns the “default value” for a type. Read more
Source§impl<T> TensorLinalgPrims<T> for HipTensorLinalgBackendwhere
T: KernelLinalgScalar,
impl<T> TensorLinalgPrims<T> for HipTensorLinalgBackendwhere
T: KernelLinalgScalar,
type Context = RocmContext
fn has_linalg_support(_op: LinalgCapabilityOp) -> bool
Source§fn solve_ex(
_ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context,
_a: &Tensor<T>,
_b: &Tensor<T>,
) -> Result<SolveTensorExResult<T>, Error>
fn solve_ex( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, _b: &Tensor<T>, ) -> Result<SolveTensorExResult<T>, Error>
Solve a square linear system while returning per-batch numerical status. Read more
fn solve( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, _b: &Tensor<T>, ) -> Result<Tensor<T>, Error>
fn lu_solve( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _factors: &Tensor<T>, _pivots: &Tensor<i32>, _b: &Tensor<T>, ) -> Result<Tensor<T>, Error>
fn solve_triangular( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, _b: &Tensor<T>, _upper: bool, ) -> Result<Tensor<T>, Error>
fn qr( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<QrTensorResult<T>, Error>
fn thin_svd( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<SvdTensorResult<T>, Error>
fn svdvals( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<Tensor<<T as LinalgScalar>::Real>, Error>
Source§fn lu_factor_ex(
_ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context,
_a: &Tensor<T>,
) -> Result<LuTensorExResult<T>, Error>
fn lu_factor_ex( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<LuTensorExResult<T>, Error>
Compute an LU factorization while returning per-batch numerical status. Read more
fn lu_factor( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<LuTensorResult<T>, Error>
Source§fn lu_factor_no_pivot(
_ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context,
_a: &Tensor<T>,
) -> Result<LuTensorResult<T>, Error>
fn lu_factor_no_pivot( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<LuTensorResult<T>, Error>
Compute an LU factorization without pivoting. Read more
Source§fn cholesky_ex(
_ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context,
_a: &Tensor<T>,
) -> Result<CholeskyTensorExResult<T>, Error>
fn cholesky_ex( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<CholeskyTensorExResult<T>, Error>
Compute a Cholesky factorization while returning per-batch numerical status. Read more
fn cholesky( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<Tensor<T>, Error>
fn eigen_sym( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<EigenTensorResult<T>, Error>
fn eig( _ctx: &mut <HipTensorLinalgBackend as TensorLinalgPrims<T>>::Context, _a: &Tensor<T>, ) -> Result<EigTensorResult<T>, Error>
impl Copy for HipTensorLinalgBackend
Auto Trait Implementations§
impl Freeze for HipTensorLinalgBackend
impl RefUnwindSafe for HipTensorLinalgBackend
impl Send for HipTensorLinalgBackend
impl Sync for HipTensorLinalgBackend
impl Unpin for HipTensorLinalgBackend
impl UnwindSafe for HipTensorLinalgBackend
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
§impl<T> DistributionExt for Twhere
T: ?Sized,
impl<T> DistributionExt for Twhere
T: ?Sized,
fn rand<T>(&self, rng: &mut (impl Rng + ?Sized)) -> Twhere
Self: Distribution<T>,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
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 moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
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