Searched refs:mat_a_size (Results 1 – 1 of 1) sorted by relevance
26 def matrix_square_root(mat_a, mat_a_size, iter_count=100, ridge_epsilon=1e-4): argument61 identity = linalg_ops.eye(math_ops.cast(mat_a_size, dtypes.int32))