Searched refs:bf16_scalar_shape (Results 1 – 1 of 1) sorted by relevance
184 Shape bf16_scalar_shape = ShapeUtil::MakeShape(BF16, {}); in TEST_F() local188 HloInstruction::CreateParameter(0, bf16_scalar_shape, "param0")); in TEST_F()190 HloInstruction::CreateParameter(1, bf16_scalar_shape, "param1")); in TEST_F()192 HloInstruction::CreateBinary(bf16_scalar_shape, HloOpcode::kAdd, in TEST_F()203 HloInstruction::CreateParameter(1, bf16_scalar_shape, "init")); in TEST_F()