Home
last modified time | relevance | path

Searched refs:diagonals_format (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_tridiag.py130 diagonals_format=_COMPACT, argument
176 diagonals_format=diagonals_format,
185 if diagonals_format not in _DIAGONAL_FORMATS:
188 ', got : {}'.format(diagonals_format))
189 if diagonals_format == _SEQUENCE:
197 self._diagonals_format = diagonals_format
209 if self.diagonals_format == _MATRIX:
211 if self.diagonals_format == _COMPACT:
227 if self.diagonals_format == _MATRIX:
229 if self.diagonals_format == _COMPACT:
[all …]
Dlinalg_impl.py450 diagonals_format='compact', argument
546 if diagonals_format == 'compact':
551 if diagonals_format == 'sequence':
584 if diagonals_format == 'matrix':
597 raise ValueError('Unrecognized diagonals_format: {}'.format(diagonals_format))
650 def tridiagonal_matmul(diagonals, rhs, diagonals_format='compact', name=None): argument
699 if diagonals_format == 'compact':
703 elif diagonals_format == 'sequence':
705 elif diagonals_format == 'matrix':
718 raise ValueError('Unrecognized diagonals_format: %s' % diagonals_format)
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_tridiag_test.py36 diagonals_format='sequence'): argument
68 if diagonals_format == 'sequence':
70 elif diagonals_format == 'compact':
72 elif diagonals_format == 'matrix':
78 if diagonals_format == 'sequence':
87 diagonals_format=diagonals_format,
116 diagonals_format='compact')
121 diag, diagonals_format='compact')
137 diagonals_format='sequence')
145 diagonals, diagonals_format='sequence')
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dtridiagonal_matmul_op_test.py68 diags_sequence, rhs, diagonals_format='sequence'),
70 diags_compact, rhs, diagonals_format='compact'),
72 diags_matrix, rhs, diagonals_format='matrix')
76 diags_sequence_batch, rhs_batch, diagonals_format='sequence'),
78 diags_compact_batch, rhs_batch, diagonals_format='compact'),
80 diags_matrix_batch, rhs_batch, diagonals_format='matrix')
156 diagonals_format='matrix')
222 diagonals_format='sequence')
Dtridiagonal_solve_op_test.py411 diagonals_format=diags_format,
649 diagonals_format="sequence",
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.-linear-operator-tridiag.pbtxt22 name: "diagonals_format"
95 …argspec: "args=[\'self\', \'diagonals\', \'diagonals_format\', \'is_non_singular\', \'is_self_adjo…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.-linear-operator-tridiag.pbtxt22 name: "diagonals_format"
95 …argspec: "args=[\'self\', \'diagonals\', \'diagonals_format\', \'is_non_singular\', \'is_self_adjo…
Dtensorflow.linalg.pbtxt261 …argspec: "args=[\'diagonals\', \'rhs\', \'diagonals_format\', \'name\'], varargs=None, keywords=No…
265 …argspec: "args=[\'diagonals\', \'rhs\', \'diagonals_format\', \'transpose_rhs\', \'conjugate_rhs\'…
/external/tensorflow/tensorflow/compiler/tests/
Dtridiagonal_solve_ops_test.py402 diagonals_format=diags_format,