Searched refs:tridiagonal_matmul (Results 1 – 5 of 5) sorted by relevance
67 linalg_impl.tridiagonal_matmul(69 linalg_impl.tridiagonal_matmul(71 linalg_impl.tridiagonal_matmul(75 linalg_impl.tridiagonal_matmul(77 linalg_impl.tridiagonal_matmul(79 linalg_impl.tridiagonal_matmul(121 linalg_impl.tridiagonal_matmul, [diags, rhs])153 result = linalg_impl.tridiagonal_matmul(220 x2 = linalg_impl.tridiagonal_matmul((upper, diag, lower),
304 return linalg.tridiagonal_matmul(
650 def tridiagonal_matmul(diagonals, rhs, diagonals_format='compact', name=None): function
260 name: "tridiagonal_matmul"
2117 * Added GPU implementation of `tf.linalg.tridiagonal_matmul`.