Searched refs:BatchNormExpanderTest (Results 1 – 1 of 1) sorted by relevance
39 class BatchNormExpanderTest : public HloTestBase { class56 TEST_F(BatchNormExpanderTest, BatchNormTraining) { in TEST_F() argument91 TEST_F(BatchNormExpanderTest, BatchNormGrad) { in TEST_F() argument133 TEST_F(BatchNormExpanderTest, BatchNormTrainingSharding) { in TEST_F() argument