Searched defs:ReduceWindowTest (Results 1 – 2 of 2) sorted by relevance
65 class ReduceWindowTest : public ::testing::WithParamInterface<bool>, class68 ReduceWindowTest() : builder_(TestName()) { set_use_bfloat16(GetParam()); } in ReduceWindowTest() function in xla::__anonacae252d0111::ReduceWindowTest106 XLA_TEST_P(ReduceWindowTest, MismatchedRanksGivesErrorStatus) { in XLA_TEST_P() argument123 XLA_TEST_P(ReduceWindowTest, R0ReduceWindow) { in XLA_TEST_P() argument135 XLA_TEST_P(ReduceWindowTest, Min3In5Stride2) { in XLA_TEST_P() argument143 XLA_TEST_P(ReduceWindowTest, Min3In5Stride2Same) { in XLA_TEST_P() argument152 XLA_TEST_P(ReduceWindowTest, Min3In5Stride1WithSamePadding) { in XLA_TEST_P() argument162 XLA_TEST_P(ReduceWindowTest, ZeroElementSmall) { in XLA_TEST_P() argument175 XLA_TEST_P(ReduceWindowTest, NonSquareSmall) { in XLA_TEST_P() argument190 XLA_TEST_P(ReduceWindowTest, MiddleDimsSmall) { in XLA_TEST_P() argument[all …]
27 class ReduceWindowTest(xla_test.XLATestCase): class