Searched defs: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
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
294 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() argument
323 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() argument
352 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() argument
[all …]
|