Home
last modified time | relevance | path

Searched defs:linalg (Results 1 – 25 of 35) sorted by relevance

12

/external/pytorch/tools/autograd/templates/
Dpython_linalg_functions.cpp53 PyObject* linalg = PyModule_Create(&def); in initLinalgFunctions() local
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dconjugate_gradient.py25 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_permutation_test.py29 linalg = linalg_lib variable
Dlinear_operator_lower_triangular_test.py25 linalg = linalg_lib variable
Dlinear_operator_toeplitz_test.py32 linalg = linalg_lib variable
Dlinear_operator_inversion_test.py26 linalg = linalg_lib variable
Dlinear_operator_householder_test.py27 linalg = linalg_lib variable
Dlinear_operator_kronecker_test.py31 linalg = linalg_lib variable
Dlinear_operator_block_lower_triangular_test.py31 linalg = linalg_lib variable
Dlinear_operator_adjoint_test.py28 linalg = linalg_lib variable
Dslicing_test.py26 linalg = linalg_lib variable
Dlinear_operator_diag_test.py27 linalg = linalg_lib variable
Dlinear_operator_composition_test.py27 linalg = linalg_lib variable
Dlinear_operator_full_matrix_test.py29 linalg = linalg_lib variable
Dlinear_operator_block_diag_test.py32 linalg = linalg_lib variable
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_toeplitz.py22 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
Dlinear_operator_tridiag.py24 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
Dlinear_operator_lower_triangular.py20 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
Dlinear_operator_permutation.py26 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
Dlinear_operator_householder.py23 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
Dlinear_operator_diag.py21 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
Dlinear_operator_zeros.py28 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
Dlinear_operator_adjoint.py20 from tensorflow.python.ops.linalg import linalg_impl as linalg unknown
/external/pytorch/torch/_refs/linalg/
D__init__.py9 import torch._refs.linalg as linalg namespace
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dembed_memref_prints.cc115 if (auto linalg = dyn_cast<linalg::LinalgOp>(op)) { in ExtractValuesToPrint() local

12