Searched refs:BFloat16NormalizationTest (Results 1 – 1 of 1) sorted by relevance
72 class BFloat16NormalizationTest : public HloTestBase { class74 BFloat16NormalizationTest() in BFloat16NormalizationTest() function in xla::BFloat16NormalizationTest92 TEST_F(BFloat16NormalizationTest, NoopIfSupported) { in TEST_F() argument120 TEST_F(BFloat16NormalizationTest, ResolveIfUnsupportedBF16) { in TEST_F() argument150 TEST_F(BFloat16NormalizationTest, ResolveUnsupportedMixedPrecisionSubtraction) { in TEST_F() argument180 TEST_F(BFloat16NormalizationTest, ResolveUnsupportedMixedPrecisionReduce) { in TEST_F() argument236 TEST_F(BFloat16NormalizationTest, ResolveMixedPrecisionTupleAllReduce) { in TEST_F() argument274 TEST_F(BFloat16NormalizationTest, ResolveMixedPrecisionTupleSort) { in TEST_F() argument304 TEST_F(BFloat16NormalizationTest, ResolveMixedPrecisionTupleSortRoot) { in TEST_F() argument341 TEST_F(BFloat16NormalizationTest, DoNotAddUnsupportedMixedPrecision) { in TEST_F() argument