Searched refs:mat_diag (Results 1 – 3 of 3) sorted by relevance
223 mat_diag = array_ops.matrix_diag_part(mat)224 self.assertEqual((3,), mat_diag.get_shape())225 self.assertAllEqual(mat_diag.eval(), v)231 mat_diag = array_ops.matrix_diag_part(mat)232 self.assertAllEqual(mat_diag.eval(), np.array([1.0, 5.0]))234 mat_diag = array_ops.matrix_diag_part(mat)235 self.assertAllEqual(mat_diag.eval(), np.array([1.0, 4.0]))
394 mat_diag = array_ops.matrix_diag_part(mat)395 new_diag = 1 + mat_diag719 mat_diag = array_ops.matrix_diag_part(mat)722 new_diag = multiplier_vector + mat_diag
2951 Output mat_diag = in TEST_F() local2953 Output mat = ops::Diag(scope.WithOpName("mat"), mat_diag); in TEST_F()