Searched refs:VariadicReduceTest (Results 1 – 1 of 1) sorted by relevance
1047 class VariadicReduceTest : public HloTestBase {}; class1049 XLA_TEST_F(VariadicReduceTest, Reduce_R3x2_to_R2x2_simple) { in XLA_TEST_F() argument1077 XLA_TEST_F(VariadicReduceTest, Reduce_R3x2_to_R1x2_simple) { in XLA_TEST_F() argument1105 XLA_TEST_F(VariadicReduceTest, Reduce_R1x2_to_R0x2_simple) { in XLA_TEST_F() argument1133 XLA_TEST_F(VariadicReduceTest, Reduce_R1x2_to_R0x2_argmax) { in XLA_TEST_F() argument1170 XLA_TEST_F(VariadicReduceTest, Reduce_R1x2_to_R0x2_argmax_column) { in XLA_TEST_F() argument1213 XLA_TEST_F(VariadicReduceTest, ReduceMultiOutputVariadicAnd) { in XLA_TEST_F() argument1245 XLA_TEST_F(VariadicReduceTest, ReduceMultiOutputVariadicDifferentLayout) { in XLA_TEST_F() argument