Searched refs:IsScalarWithElementType (Results 1 – 5 of 5) sorted by relevance
29 ShapeUtil::IsScalarWithElementType(instruction->shape(), F32)) { in IsConstantR0F32()
141 EXPECT_TRUE(ShapeUtil::IsScalarWithElementType(parameter->shape(), F32)); in TEST_F()142 EXPECT_FALSE(ShapeUtil::IsScalarWithElementType(parameter->shape(), S32)); in TEST_F()
79 ShapeUtil::IsScalarWithElementType(TypeToShape(b.getF32Type()), F32)); in TEST()
321 static bool IsScalarWithElementType(const Shape& shape,
465 /* static */ bool ShapeUtil::IsScalarWithElementType( in IsScalarWithElementType() function in xla::ShapeUtil