Searched refs:AutoMixedPrecisionTest (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 97 class AutoMixedPrecisionTest : public GrapplerTest { class
209 TEST_F(AutoMixedPrecisionTest, NoOp) { in TEST_F() argument
246 TEST_F(AutoMixedPrecisionTest, AlreadyFp16) { in TEST_F() argument
284 TEST_F(AutoMixedPrecisionTest, Simple) { in TEST_F() argument
333 TEST_F(AutoMixedPrecisionTest, BidirectionalClearChain) { in TEST_F() argument
372 TEST_F(AutoMixedPrecisionTest, PreserveFetches) { in TEST_F() argument
417 TEST_F(AutoMixedPrecisionTest, PreserveCPUNodes) { in TEST_F() argument
458 TEST_F(AutoMixedPrecisionTest, PreserveIdentityAfterVariable) { in TEST_F() argument
502 TEST_F(AutoMixedPrecisionTest, FusedBatchNorm) { in TEST_F() argument
561 TEST_F(AutoMixedPrecisionTest, RepeatedAndListTypeAttrs) { in TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | auto_mixed_precision_test.py | 330 class AutoMixedPrecisionTest(test.TestCase, parameterized.TestCase): class
337 super(AutoMixedPrecisionTest, self).setUp()
348 super(AutoMixedPrecisionTest, self).tearDown()
|