Searched refs:mat_m (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | matrix_functions.py | 106 def mat_power(mat_m, p): argument
121 power = math_ops.matmul(mat_m, power) if power is not None else mat_m
123 mat_m = math_ops.matmul(mat_m, mat_m)
126 def _iter_condition(i, mat_m, _): argument
129 math_ops.reduce_max(math_ops.abs(mat_m - identity)) > epsilon)
131 def _iter_body(i, mat_m, mat_x): argument
132 mat_m_i = (1 - alpha) * identity + alpha * mat_m
133 return (i + 1, math_ops.matmul(mat_power(mat_m_i, -1.0 / alpha), mat_m),
|