Home
last modified time | relevance | path

Searched full:linalg_inv_ex (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DInverse.mm9 #include <ATen/ops/linalg_inv_ex.h>
20 …"torch.linalg_inv_ex.inverse is supported by MPS on MacOS 13+, please upgrade. Falling back to CPU…
/aosp_15_r20/external/pytorch/test/
H A Dtest_namedtuple_return_api.py17 'qr', 'geqrf', 'slogdet', 'sort', 'topk', 'linalg_inv_ex',
95 op(operators=['linalg_inv_ex'], input=(), names=('inverse', 'info'), hasout=True),
/aosp_15_r20/external/executorch/kernels/aten/
H A Dfunctions.yaml214 - op: linalg_inv_ex.inverse
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect872 aten::linalg_inv_ex
873 aten::linalg_inv_ex.inverse
/aosp_15_r20/external/pytorch/test/edge/
H A Dselected_operators.yaml212 aten::linalg_inv_ex.inverse:
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dannotated_ops535 linalg_inv_ex, alias
/aosp_15_r20/external/pytorch/test/lazy/
H A Dtest_ts_opinfo.py85 "linalg_inv_ex",
/aosp_15_r20/external/pytorch/docs/source/
H A Damp.rst492 ``linalg_inv_ex``
H A Dconf.py3109 "linalg_inv_ex",
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp453 KERNEL_CPU(linalg_inv_ex, fp32) in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesLinearAlgebra.cpp736 LINALG_CHECK_MATRIX_UNARY_TWO_OUT(linalg_inv_ex, linalg.inv_ex);
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp86 #include <ATen/ops/linalg_inv_ex.h>
3146 std::tie(inverse, info) = at::linalg_inv_ex(self); in _linalg_cond_helper()
3300 …auto [result, info] = at::linalg_inv_ex(self.reshape_symint({prod_ind_end, prod_ind_end}), /*check… in linalg_tensorinv()
H A DBatchLinearAlgebra.cpp71 #include <ATen/ops/linalg_inv_ex.h>
546 TORCH_META_FUNC(linalg_inv_ex)(const Tensor& A, bool check_errors) { in TORCH_META_FUNC() argument
1609 auto [result, info] = at::linalg_inv_ex(A); in linalg_inv()
H A Dnative_functions.yaml14001 - func: linalg_inv_ex(Tensor A, *, bool check_errors=False) -> (Tensor inverse, Tensor info)
14003 structured_delegate: linalg_inv_ex.inverse
14005 - func: linalg_inv_ex.inverse(Tensor A, *, bool check_errors=False, Tensor(a!) inverse, Tensor(b!) …
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_variable_type.py302 "linalg_inv_ex",
H A Dderivatives.yaml886 - name: linalg_inv_ex(Tensor A, *, bool check_errors=False) -> (Tensor inverse, Tensor info)
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/
H A Dlinalg.py1973 aten_name="linalg_inv_ex",
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py911 "torch._C._linalg.linalg_inv_ex",
/aosp_15_r20/external/pytorch/torch/linalg/
H A D__init__.py311 inv_ex = _add_docstr(_linalg.linalg_inv_ex, r"""
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2317 make_fallback(aten.linalg_inv_ex)
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py1006 @register_meta(aten.linalg_inv_ex.default)