Don't remove all aliases from linalg namespace.
Get rid of redundant aliases. PiperOrigin-RevId: 168257658
This commit is contained in:
parent
509372c2ee
commit
e15f4cae23
@ -30,7 +30,6 @@ from tensorflow.python.ops import special_math_ops
|
||||
# pylint: disable=wildcard-import,unused-import
|
||||
from tensorflow.python.ops.linalg_impl import *
|
||||
# pylint: enable=wildcard-import
|
||||
from tensorflow.python.util.all_util import remove_undocumented
|
||||
|
||||
# Linear algebra ops.
|
||||
band_part = array_ops.matrix_band_part
|
||||
@ -55,21 +54,13 @@ trace = math_ops.trace
|
||||
transpose = array_ops.matrix_transpose
|
||||
triangular_solve = linalg_ops.matrix_triangular_solve
|
||||
|
||||
# Additional aliases for consistency with root namespace for ops whose names
|
||||
# changed significantly when moving to the linalg namespace.
|
||||
self_adjoint_eig = linalg_ops.self_adjoint_eig
|
||||
self_adjoint_eigvals = linalg_ops.self_adjoint_eigvals
|
||||
solve_ls = linalg_ops.matrix_solve_ls
|
||||
|
||||
# Seal API.
|
||||
del absolute_import
|
||||
del array_ops
|
||||
del division
|
||||
del gen_linalg_ops
|
||||
del linalg_ops
|
||||
del math_ops
|
||||
del ops
|
||||
del print_function
|
||||
del special_math_ops
|
||||
|
||||
_allowed_symbols = []
|
||||
|
||||
remove_undocumented(__name__, _allowed_symbols)
|
||||
|
@ -1,7 +1,91 @@
|
||||
path: "tensorflow.linalg"
|
||||
tf_module {
|
||||
member_method {
|
||||
name: "band_part"
|
||||
argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "cholesky"
|
||||
argspec: "args=[\'input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "cholesky_solve"
|
||||
argspec: "args=[\'chol\', \'rhs\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "det"
|
||||
argspec: "args=[\'input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "diag"
|
||||
argspec: "args=[\'diagonal\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "diag_part"
|
||||
argspec: "args=[\'input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "eigh"
|
||||
argspec: "args=[\'tensor\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "eigvalsh"
|
||||
argspec: "args=[\'tensor\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "einsum"
|
||||
argspec: "args=[\'equation\'], varargs=inputs, keywords=kwargs, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "eye"
|
||||
argspec: "args=[\'num_rows\', \'num_columns\', \'batch_shape\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \"<dtype: \'float32\'>\", \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "inv"
|
||||
argspec: "args=[\'input\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "logdet"
|
||||
argspec: "args=[\'matrix\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "lstsq"
|
||||
argspec: "args=[\'matrix\', \'rhs\', \'l2_regularizer\', \'fast\', \'name\'], varargs=None, keywords=None, defaults=[\'0.0\', \'True\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "norm"
|
||||
argspec: "args=[\'tensor\', \'ord\', \'axis\', \'keep_dims\', \'name\'], varargs=None, keywords=None, defaults=[\'euclidean\', \'None\', \'False\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "qr"
|
||||
argspec: "args=[\'input\', \'full_matrices\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "set_diag"
|
||||
argspec: "args=[\'input\', \'diagonal\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "solve"
|
||||
argspec: "args=[\'matrix\', \'rhs\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "svd"
|
||||
argspec: "args=[\'tensor\', \'full_matrices\', \'compute_uv\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'True\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "tensordot"
|
||||
argspec: "args=[\'a\', \'b\', \'axes\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "trace"
|
||||
argspec: "args=[\'x\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "transpose"
|
||||
argspec: "args=[\'a\', \'name\'], varargs=None, keywords=None, defaults=[\'matrix_transpose\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "triangular_solve"
|
||||
argspec: "args=[\'matrix\', \'rhs\', \'lower\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'True\', \'False\', \'None\'], "
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user