# limitations under the License.
# ============================================================================
"""Numpy implementations of `tf.linalg` functions."""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import collections
# Dependency imports
import numpy as np
from tensorflow_probability.python.internal.backend.numpy import _utils as utils
from tensorflow_probability.python.internal.backend.numpy import ops
scipy_linalg = utils.try_import('scipy.linalg')
__all__ = [
'adjoint',
'band_part',
'cholesky',
'cholesky_solve',
'det',
'diag',
'diag_part',
'einsum',
'eye',
'inv',
'lu',
'matmul',
'matvec',
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import functools
# Dependency imports
import numpy as np
import tensorflow.compat.v2 as tf
from tensorflow_probability.python.internal.backend.numpy import _utils as utils
from tensorflow_probability.python.internal.backend.numpy.numpy_array import _reverse
scipy_special = utils.try_import('scipy.special')
JAX_MODE = False
if JAX_MODE:
import jax # pylint: disable=g-import-not-at-top
__all__ = [
'abs',
'accumulate_n',
'acos',
'acosh',
'add',
'add_n',
