Searched refs:identity_operator (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | cholesky_registrations.py | 56 def _cholesky_identity(identity_operator): argument
58 num_rows=identity_operator._num_rows, # pylint: disable=protected-access
59 batch_shape=identity_operator.batch_shape,
60 dtype=identity_operator.dtype,
69 def _cholesky_scaled_identity(identity_operator): argument
71 num_rows=identity_operator._num_rows, # pylint: disable=protected-access
72 multiplier=math_ops.sqrt(identity_operator.multiplier),
|
| D | adjoint_registrations.py | 53 def _adjoint_identity(identity_operator): argument
54 return identity_operator
59 def _adjoint_scaled_identity(identity_operator): argument
60 multiplier = identity_operator.multiplier
65 num_rows=identity_operator._num_rows, # pylint: disable=protected-access
67 is_non_singular=identity_operator.is_non_singular,
68 is_self_adjoint=identity_operator.is_self_adjoint,
69 is_positive_definite=identity_operator.is_positive_definite,
|
| D | inverse_registrations.py | 67 def _inverse_identity(identity_operator): argument
68 return identity_operator
73 def _inverse_scaled_identity(identity_operator): argument
75 num_rows=identity_operator._num_rows, # pylint: disable=protected-access
76 multiplier=1. / identity_operator.multiplier,
77 is_non_singular=identity_operator.is_non_singular,
79 is_positive_definite=identity_operator.is_positive_definite,
|