/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
H A D | Inverse.mm | 9 #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 D | test_namedtuple_return_api.py | 17 '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 D | functions.yaml | 214 - op: linalg_inv_ex.inverse
|
/aosp_15_r20/external/pytorch/test/expect/ |
H A D | HasDecompTest.test_has_decomposition.expect | 872 aten::linalg_inv_ex 873 aten::linalg_inv_ex.inverse
|
/aosp_15_r20/external/pytorch/test/edge/ |
H A D | selected_operators.yaml | 212 aten::linalg_inv_ex.inverse:
|
/aosp_15_r20/external/pytorch/functorch/op_analysis/ |
H A D | annotated_ops | 535 linalg_inv_ex, alias
|
/aosp_15_r20/external/pytorch/test/lazy/ |
H A D | test_ts_opinfo.py | 85 "linalg_inv_ex",
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | amp.rst | 492 ``linalg_inv_ex``
|
H A D | conf.py | 3109 "linalg_inv_ex",
|
/aosp_15_r20/external/pytorch/aten/src/ATen/ |
H A D | autocast_mode.cpp | 453 KERNEL_CPU(linalg_inv_ex, fp32) in TORCH_LIBRARY_IMPL()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
H A D | BatchRulesLinearAlgebra.cpp | 736 LINALG_CHECK_MATRIX_UNARY_TWO_OUT(linalg_inv_ex, linalg.inv_ex);
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | LinearAlgebra.cpp | 86 #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 D | BatchLinearAlgebra.cpp | 71 #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 D | native_functions.yaml | 14001 - 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 D | gen_variable_type.py | 302 "linalg_inv_ex",
|
H A D | derivatives.yaml | 886 - 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 D | linalg.py | 1973 aten_name="linalg_inv_ex",
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | trace_rules.py | 911 "torch._C._linalg.linalg_inv_ex",
|
/aosp_15_r20/external/pytorch/torch/linalg/ |
H A D | __init__.py | 311 inv_ex = _add_docstr(_linalg.linalg_inv_ex, r"""
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | lowering.py | 2317 make_fallback(aten.linalg_inv_ex)
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _meta_registrations.py | 1006 @register_meta(aten.linalg_inv_ex.default)
|