Searched full:linalg_inv_ex (Results 1 – 21 of 21) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| 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…
|
| /external/pytorch/test/ |
| 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),
|
| /external/executorch/kernels/aten/ |
| D | functions.yaml | 214 - op: linalg_inv_ex.inverse
|
| /external/pytorch/test/edge/ |
| D | selected_operators.yaml | 212 aten::linalg_inv_ex.inverse:
|
| /external/pytorch/functorch/op_analysis/ |
| D | annotated_ops | 535 linalg_inv_ex, alias
|
| /external/pytorch/test/lazy/ |
| D | test_ts_opinfo.py | 85 "linalg_inv_ex",
|
| /external/pytorch/docs/source/ |
| D | amp.rst | 492 ``linalg_inv_ex``
|
| D | conf.py | 3109 "linalg_inv_ex",
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_has_decomposition.expect | 872 aten::linalg_inv_ex
873 aten::linalg_inv_ex.inverse
|
| /external/pytorch/aten/src/ATen/ |
| D | autocast_mode.cpp | 453 KERNEL_CPU(linalg_inv_ex, fp32) in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesLinearAlgebra.cpp | 736 LINALG_CHECK_MATRIX_UNARY_TWO_OUT(linalg_inv_ex, linalg.inv_ex);
|
| /external/pytorch/aten/src/ATen/native/ |
| 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()
|
| 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()
|
| 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!) …
|
| /external/pytorch/tools/autograd/ |
| D | gen_variable_type.py | 302 "linalg_inv_ex",
|
| D | derivatives.yaml | 886 - name: linalg_inv_ex(Tensor A, *, bool check_errors=False) -> (Tensor inverse, Tensor info)
|
| /external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| D | linalg.py | 1973 aten_name="linalg_inv_ex",
|
| /external/pytorch/torch/_dynamo/ |
| D | trace_rules.py | 911 "torch._C._linalg.linalg_inv_ex",
|
| /external/pytorch/torch/linalg/ |
| D | __init__.py | 311 inv_ex = _add_docstr(_linalg.linalg_inv_ex, r"""
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 2317 make_fallback(aten.linalg_inv_ex)
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 1006 @register_meta(aten.linalg_inv_ex.default)
|