Searched refs:TestMatrixDiagonal (Results 1 – 1 of 1) sorted by relevance
40 void TestMatrixDiagonal();77 void MatrixTest::TestMatrixDiagonal() { in TestMatrixDiagonal() function in xla::__anon94e7bccf0111::MatrixTest120 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_S32) { TestMatrixDiagonal<int32>(); } in XLA_TEST_F()122 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_S64) { TestMatrixDiagonal<int64>(); } in XLA_TEST_F()124 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_F32) { TestMatrixDiagonal<float>(); } in XLA_TEST_F()