Searched refs:mat_diag (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 780 mat_diag = array_ops.matrix_diag_part(mat)
781 self.assertEqual((3,), mat_diag.get_shape())
782 self.assertAllEqual(mat_diag, v)
786 mat_diag = array_ops.matrix_diag_part(mat, k=offset)
787 self.assertEqual((3,), mat_diag.get_shape())
788 self.assertAllEqual(mat_diag, v)
795 mat_diag = array_ops.matrix_diag_part(mat[0], k=diags, align=align)
796 self.assertEqual(mat_diag.get_shape(), solution[0].shape)
797 self.assertAllEqual(mat_diag, solution[0])
803 mat_diag = array_ops.matrix_diag_part(mat)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_identity.py | 410 mat_diag = array_ops.matrix_diag_part(mat)
411 new_diag = 1 + mat_diag
749 mat_diag = array_ops.matrix_diag_part(mat)
752 new_diag = multiplier_vector + mat_diag
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 3297 Output mat_diag = in TEST_F() local
3299 Output mat = ops::Diag(scope.WithOpName("mat"), mat_diag); in TEST_F()
|