Searched refs:fusion0 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_test.cc | 3319 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P() local
3323 HloInstruction::CreateGetTupleElement(shape, fusion0, 0)); in TEST_P()
3325 HloInstruction::CreateGetTupleElement(shape, fusion0, 1)); in TEST_P()
3346 {fusion0}, fusion1_computation)); in TEST_P()
3355 {p0, fusion0, element0, element1, negate0, negate1, negate2, negate3, in TEST_P()
3403 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P() local
3422 {fusion0}, fusion1_computation)); in TEST_P()
3428 computation, {p0, fusion0, negate0, negate1, negate2, negate3, negate4, in TEST_P()
3478 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P() local
3483 {fusion0}, fusion1_computation)); in TEST_P()
[all …]
|
| D | bfloat16_propagation_test.cc | 398 auto fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F() local
412 dot->shape(), HloInstruction::FusionKind::kCustom, {fusion0}, comp_f1)); in TEST_F()
|