Home
last modified time | relevance | path

Searched refs:TestMatrixDiagonal (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix_test.cc40 void TestMatrixDiagonal();
77 void MatrixTest::TestMatrixDiagonal() { in TestMatrixDiagonal() function in xla::__anon94e7bccf0111::MatrixTest
120 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()