Searched defs:diagonals_format (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tridiagonal_matmul_ops_test.py | 35 def _jit_tridiagonal_matmul(self, diagonals, rhs, diagonals_format): argument
40 def _tridiagonal_matmul(self, diagonals, rhs, diagonals_format): argument
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_tridiag.py | 127 diagonals_format=_COMPACT, argument
381 def diagonals_format(self): member in LinearOperatorTridiag
|
| D | linalg_impl.py | 447 diagonals_format='compact', argument
669 def tridiagonal_matmul(diagonals, rhs, diagonals_format='compact', name=None): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_tridiag_test.py | 33 diagonals_format='sequence'): argument
|