Searched refs:RunR2ToR1Test (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 229 void RunR2ToR1Test(int64 rows, int64 cols, int64 minor = 1, int64 major = 0) { in RunR2ToR1Test() function in xla::__anon10c8b2690111::ReduceTest
392 XLA_TEST_F(ReduceTest, ReduceR2_0x2_To_R1) { RunR2ToR1Test(0, 2); } in XLA_TEST_F()
393 XLA_TEST_F(ReduceTest, ReduceR2_1x1_To_R1) { RunR2ToR1Test(1, 1); } in XLA_TEST_F()
396 XLA_TEST_F(ReduceTest, ReduceR2_2x2_To_R1) { RunR2ToR1Test(2, 2); } in XLA_TEST_F()
397 XLA_TEST_F(ReduceTest, ReduceR2_8x8_To_R1) { RunR2ToR1Test(8, 8); } in XLA_TEST_F()
398 XLA_TEST_F(ReduceTest, ReduceR2_9x9_To_R1) { RunR2ToR1Test(9, 9); } in XLA_TEST_F()
399 XLA_TEST_F(ReduceTest, ReduceR2_50x111_To_R1) { RunR2ToR1Test(50, 111); } in XLA_TEST_F()
400 XLA_TEST_F(ReduceTest, ReduceR2_111x50_To_R1) { RunR2ToR1Test(111, 50); } in XLA_TEST_F()
402 RunR2ToR1Test(111, 50, 0, 1); in XLA_TEST_F()
404 XLA_TEST_F(ReduceTest, ReduceR2_1024x1024_To_R1) { RunR2ToR1Test(1024, 1024); } in XLA_TEST_F()
[all …]
|