Lines Matching refs:element_type
101 EXPECT_EQ(add0->shape().element_type(), BF16);
102 EXPECT_EQ(add1->shape().element_type(), F32);
130 EXPECT_EQ(mul0->shape().element_type(), F32);
131 EXPECT_EQ(mul1->shape().element_type(), F32);
160 EXPECT_EQ(sub0->shape().element_type(), F32);
161 EXPECT_EQ(sub1->shape().element_type(), F32);
202 .element_type(),
207 .element_type(),
212 .element_type(),
214 EXPECT_EQ(reduce->shape().element_type(), F32);
216 EXPECT_EQ(input->shape().element_type(), F32);
218 EXPECT_EQ(reduce->operand(1)->shape().element_type(), F32);
243 EXPECT_EQ(gte->shape().element_type(), BF16);
244 EXPECT_EQ(crs->operand(1)->shape().element_type(), F32);
245 EXPECT_EQ(ShapeUtil::GetSubshape(crs->shape(), {1}).element_type(), F32);
