bfloat16_propagation_test.cc | 438 HloInstruction* a_f = local 443 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a_f, b_f)); 477 HloInstruction* a_f = local 482 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a_f, b_f)); 484 builder_f.AddInstruction(HloInstruction::CreateTuple({a_f, add_f})); 503 EXPECT_FALSE(OutputsBF16(a_f)); 657 HloInstruction* a_f = local 659 builder_f.AddInstruction(HloInstruction::CreateTranspose(shape, a_f, {0, 1})); 677 EXPECT_FALSE(OutputsBF16(a_f)); [all...] |