Searched refs:matrix_diag_part (Results 1 – 25 of 33) sorted by relevance
12
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixDiagPartV3.pbtxt | 93 tf.matrix_diag_part(input) ==> [[1, 6, 7], # Output shape: (2, 3)
97 tf.matrix_diag_part(input, k = 1)
102 tf.matrix_diag_part(input, k = (-1, 2))
113 tf.matrix_diag_part(input, k = (-1, 2), align="LEFT_RIGHT")
124 tf.matrix_diag_part(input, k = (-2, -1))
131 tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
|
| D | api_def_MatrixDiagPartV2.pbtxt | 71 tf.matrix_diag_part(input) ==> [[1, 6, 7], # Output shape: (2, 3)
75 tf.matrix_diag_part(input, k = 1)
80 tf.matrix_diag_part(input, k = (-1, 1))
89 tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
|
| D | api_def_MatrixDiagPart.pbtxt | 42 tf.matrix_diag_part(input) ==> [[1, 2, 3, 4], [5, 6, 7, 8]]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 780 mat_diag = array_ops.matrix_diag_part(mat)
786 mat_diag = array_ops.matrix_diag_part(mat, k=offset)
795 mat_diag = array_ops.matrix_diag_part(mat[0], k=diags, align=align)
803 mat_diag = array_ops.matrix_diag_part(mat)
806 mat_diag = array_ops.matrix_diag_part(mat)
814 mat_diag = array_ops.matrix_diag_part(
826 mat_batch_diag = array_ops.matrix_diag_part(mat_batch)
836 mat_batch_diag = array_ops.matrix_diag_part(
861 mat_batch_diag = array_ops.matrix_diag_part(mat_batch)
871 mat_batch_diag = array_ops.matrix_diag_part(
[all …]
|
| D | cholesky_op_test.py | 46 0.5 * array_ops.matrix_diag_part(middle))
60 0.5 * array_ops.matrix_diag_part(middle))
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatrixDiagPartV3.pbtxt | 7 name: "matrix_diag_part"
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_householder_test.py | 74 matrix, 1. + array_ops.matrix_diag_part(matrix))
|
| D | linear_operator_lower_triangular_test.py | 50 tril = array_ops.matrix_diag_part(tril)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 371 return array_ops.matrix_diag_part(grad)
376 return array_ops.matrix_diag_part(
382 return array_ops.matrix_diag_part(
447 grad_diag = array_ops.matrix_diag_part(grad)
483 grad_diag = array_ops.matrix_diag_part(grad, k=op.inputs[2])
523 grad_diag = array_ops.matrix_diag_part(grad, k=op.inputs[2], align=align)
|
| D | linalg_grad.py | 477 0.5 * array_ops.matrix_diag_part(middle))
660 grad_a = array_ops.matrix_diag_part(
663 grad_a = array_ops.matrix_diag_part(
748 array_ops.matrix_diag_part(
806 0.5 * array_ops.matrix_diag_part(grad_a))
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_lower_triangular.py | 183 return array_ops.matrix_diag_part(self._tril)
|
| D | linear_operator.py | 572 array_ops.matrix_diag_part(linalg_ops.cholesky(self.to_dense())),
761 diag = array_ops.matrix_diag_part(linalg_ops.cholesky(self.to_dense()))
1026 return array_ops.matrix_diag_part(self.to_dense())
|
| D | linalg_impl.py | 46 diag_part = array_ops.matrix_diag_part
98 math_ops.log(math_ops.real(array_ops.matrix_diag_part(chol))),
592 diagonals = array_ops.matrix_diag_part(
712 diags = array_ops.matrix_diag_part(
|
| D | linear_operator_diag.py | 257 x_diag = array_ops.matrix_diag_part(x)
|
| D | linear_operator_identity.py | 410 mat_diag = array_ops.matrix_diag_part(mat)
749 mat_diag = array_ops.matrix_diag_part(mat)
|
| D | linear_operator_householder.py | 248 matrix, 1. + array_ops.matrix_diag_part(matrix))
|
| D | linear_operator_tridiag.py | 345 return array_ops.matrix_diag_part(self.diagonals)
|
| D | linear_operator_low_rank_update.py | 405 chol_cap_diag = array_ops.matrix_diag_part(
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | dirichlet_test.py | 175 sample_var = array_ops.matrix_diag_part(sample_cov)
|
| D | multinomial_test.py | 280 sample_var = array_ops.matrix_diag_part(sample_cov)
|
| D | dirichlet_multinomial_test.py | 245 sample_var = array_ops.matrix_diag_part(sample_cov)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_diag_ops_test.py | 674 output = array_ops.matrix_diag_part(**params)
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 310 result = array_ops.matrix_diag_part(v, k=k)
326 return array_ops.matrix_diag_part(a)
345 a = array_ops.matrix_diag_part(a, k=offset)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 393 return array_ops.matrix_diag_part(
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 577 diag = array_ops.matrix_diag_part(matrix)
|
12