Searched refs:ShapeIsCompatible (Results 1 – 3 of 3) sorted by relevance
33 EXPECT_TRUE(IrArray::Index::ShapeIsCompatible(a, b)); in TEST()34 EXPECT_TRUE(IrArray::Index::ShapeIsCompatible(a, c)); in TEST()35 EXPECT_FALSE(IrArray::Index::ShapeIsCompatible(a, d)); in TEST()36 EXPECT_FALSE(IrArray::Index::ShapeIsCompatible(a, e)); in TEST()37 EXPECT_FALSE(IrArray::Index::ShapeIsCompatible(a, f)); in TEST()
113 static bool ShapeIsCompatible(const Shape& a, const Shape& b);115 bool ShapeIsCompatible(const Shape& a) const { in ShapeIsCompatible() function116 return ShapeIsCompatible( in ShapeIsCompatible()
452 CHECK(index.ShapeIsCompatible(shape_)); in EmitArrayElementAddress()531 bool IrArray::Index::ShapeIsCompatible(const Shape& a, const Shape& b) { in ShapeIsCompatible() function in xla::llvm_ir::IrArray::Index