Searched refs:fusion1 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 267 auto fusion1 = result->operand(0); in TEST_F() local
269 EXPECT_EQ(HloOpcode::kFusion, fusion1->opcode()); in TEST_F()
274 EXPECT_EQ(4, fusion1->fused_instruction_count()); in TEST_F()
278 EXPECT_EQ(0, fusion1->operand_count()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_test.cc | 3344 HloInstruction* fusion1 = builder.AddInstruction( in TEST_P() local
3348 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, add1, fusion1)); in TEST_P()
3356 negate4, negate5, negate6, add0, add1, fusion1, mul}); in TEST_P()
3360 EXPECT_THAT(fusion1, in TEST_P()
3420 HloInstruction* fusion1 = builder.AddInstruction( in TEST_P() local
3429 negate5, negate6, fusion1}); in TEST_P()
3435 fusion1, in TEST_P()
3481 HloInstruction* fusion1 = builder.AddInstruction( in TEST_P() local
3488 schedule.set_sequence(computation, {p0, fusion0, fusion1}); in TEST_P()
3492 EXPECT_THAT(fusion1, op::Fusion(op::Fusion())); in TEST_P()
|
| D | bfloat16_propagation_test.cc | 411 auto fusion1 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F() local
418 EXPECT_EQ(computation->root_instruction(), fusion1); in TEST_F()
|