Searched defs:ReduceWindowTest (Results 1 – 2 of 2) sorted by relevance
67 class ReduceWindowTest : public ::testing::WithParamInterface<bool>, class70 ReduceWindowTest() : builder_(TestName()) { set_use_bfloat16(GetParam()); } in ReduceWindowTest() function in xla::__anonf63bc5e00111::ReduceWindowTest108 TEST_P(ReduceWindowTest, MismatchedRanksGivesErrorStatus) { in TEST_P() argument125 TEST_P(ReduceWindowTest, R0ReduceWindow) { in TEST_P() argument137 TEST_P(ReduceWindowTest, Min3In5Stride2) { in TEST_P() argument145 TEST_P(ReduceWindowTest, Min3In5Stride1WithSamePadding) { in TEST_P() argument155 XLA_TEST_P(ReduceWindowTest, ZeroElementSmall) { in XLA_TEST_P() argument168 TEST_P(ReduceWindowTest, NonSquareSmall) { in TEST_P() argument183 TEST_P(ReduceWindowTest, MiddleDimsSmall) { in TEST_P() argument197 TEST_P(ReduceWindowTest, Along2ndMinorDim) { in TEST_P() argument[all …]
31 class ReduceWindowTest(xla_test.XLATestCase): class