Searched refs:TestMatrixDiagonal (Results 1 – 1 of 1) sorted by relevance
34 void TestMatrixDiagonal();52 void MatrixTest::TestMatrixDiagonal() { in TestMatrixDiagonal() function in xla::__anon6a5891b80111::MatrixTest76 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_S32) { TestMatrixDiagonal<int32>(); } in XLA_TEST_F()78 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_S64) { TestMatrixDiagonal<int64>(); } in XLA_TEST_F()80 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_F32) { TestMatrixDiagonal<float>(); } in XLA_TEST_F()