Searched defs:InstructionFusionTest (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion_test.cc | 32 using InstructionFusionTest = HloTestBase; typedef
34 TEST_F(InstructionFusionTest, in TEST_F() argument
54 TEST_F(InstructionFusionTest, in TEST_F() argument
74 TEST_F(InstructionFusionTest, in TEST_F() argument
93 TEST_F(InstructionFusionTest, in TEST_F() argument
112 TEST_F(InstructionFusionTest, PotentialBitcastReshapeOfDotFused) { in TEST_F() argument
131 TEST_F(InstructionFusionTest, PotentialBitcastTransposeOfDotUnfused) { in TEST_F() argument
149 TEST_F(InstructionFusionTest, BroadcastIntoReduce) { in TEST_F() argument
178 TEST_F(InstructionFusionTest, DoNotFuseLayoutChangingOpWithReduce) { in TEST_F() argument
201 TEST_F(InstructionFusionTest, DoNotFuseLayoutChangingOpWithReduceFusion) { in TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | instruction_fusion_test.cc | 26 using InstructionFusionTest = HloTestBase; typedef
48 TEST_F(InstructionFusionTest, FuseInstructions) { in TEST_F() argument
68 TEST_F(InstructionFusionTest, FuseIntoFusionInstruction) { in TEST_F() argument
91 TEST_F(InstructionFusionTest, FuseInstructionsIntoMultiOutput) { in TEST_F() argument
112 TEST_F(InstructionFusionTest, AvoidDuplicationIfNotAllFusible) { in TEST_F() argument
151 TEST_F(InstructionFusionTest, FuseCheapNonDuplicatableOps) { in TEST_F() argument
172 TEST_F(InstructionFusionTest, AvoidDuplicationIfNotAllFusibleRecursively) { in TEST_F() argument
313 TEST_F(InstructionFusionTest, AllowUnaryDuplication) { in TEST_F() argument
336 TEST_F(InstructionFusionTest, AllowEffectiveUnaryDuplication) { in TEST_F() argument
364 TEST_F(InstructionFusionTest, AllowBinarySameValueOperandsDuplication) { in TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 36 using InstructionFusionTest = HloTestBase; typedef
50 TEST_F(InstructionFusionTest, DotOperationFusion_Basic_0) { in TEST_F() argument
69 TEST_F(InstructionFusionTest, DotOperationFusion_Basic_1) { in TEST_F() argument
88 TEST_F(InstructionFusionTest, DotOperationFusion_Bitcast) { in TEST_F() argument
109 TEST_F(InstructionFusionTest, DotOperationFusion_Reshape) { in TEST_F() argument
131 TEST_F(InstructionFusionTest, DotOperationFusion_TooLarge) { in TEST_F() argument
150 TEST_F(InstructionFusionTest, DotOperationFusion_ElementReuse) { in TEST_F() argument
169 TEST_F(InstructionFusionTest, DotOperationFusion_TransposeFusion_RHS) { in TEST_F() argument
193 TEST_F(InstructionFusionTest, DotOperationFusion_TransposeFusion_LHS) { in TEST_F() argument
217 TEST_F(InstructionFusionTest, in TEST_F() argument
[all …]
|