Searched refs:MultiOutputFusionTest (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | multi_output_fusion_test.cc | 31 using MultiOutputFusionTest = HloTestBase; typedef 47 TEST_F(MultiOutputFusionTest, MultiOutputFusionSiblingReduceAndReduceFusion) { in TEST_F() argument 76 TEST_F(MultiOutputFusionTest, MultiOutputFusionDifferentReduceInputShapes) { in TEST_F() argument 103 TEST_F(MultiOutputFusionTest, MultiOutputFusionDifferentReduceOutputShapes) { in TEST_F() argument 130 TEST_F(MultiOutputFusionTest, MultiOutputFusionSiblingReduceFusions) { in TEST_F() argument 164 TEST_F(MultiOutputFusionTest, in TEST_F() argument 197 TEST_F(MultiOutputFusionTest, in TEST_F() argument 230 TEST_F(MultiOutputFusionTest, MultiOutputFusionTwoLoops) { in TEST_F() argument 259 TEST_F(MultiOutputFusionTest, MultiOutputFusionLoopReduceToInputFusion) { in TEST_F() argument 279 TEST_F(MultiOutputFusionTest, MultiOutputFusionLoopElementwise) { in TEST_F() argument [all …]
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | multioutput_fusion_test.cc | 50 class MultiOutputFusionTest : public HloTestBase { class 52 MultiOutputFusionTest() { error_spec_ = ErrorSpec{0.0001, 1e-2}; } in MultiOutputFusionTest() function in xla::__anon0e8629770111::MultiOutputFusionTest 187 XLA_TEST_F(MultiOutputFusionTest, 2DNofusion) { RunTest2D(false, 5); } 188 XLA_TEST_F(MultiOutputFusionTest, 2DFusion) { RunTest2D(true, 5); } 189 XLA_TEST_F(MultiOutputFusionTest, 2DFusionSize129) { RunTest2D(true, 129); } 190 XLA_TEST_F(MultiOutputFusionTest, DiffentTypesNoFusion) { RunTest1D(false, 8); } in XLA_TEST_F() argument 191 XLA_TEST_F(MultiOutputFusionTest, DiffentTypesFusion) { RunTest1D(true, 8); } in XLA_TEST_F() argument 193 XLA_TEST_F(MultiOutputFusionTest, FusionNodeIsRoot) { in XLA_TEST_F() argument 223 XLA_TEST_F(MultiOutputFusionTest, MultiOutputLoopFusion) { in XLA_TEST_F() argument 248 XLA_TEST_F(MultiOutputFusionTest, MultiOutputLoopFeedingMap) { in XLA_TEST_F() argument [all …]
|