Home
last modified time | relevance | path

Searched refs:getValueAsShape (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/unittests/Interfaces/
H A DInferTypeOpInterfaceTest.cpp57 EXPECT_FALSE(range.getValueAsShape(0)); in TEST_F()
58 ASSERT_TRUE(range.getValueAsShape(1)); in TEST_F()
59 EXPECT_TRUE(range.getValueAsShape(1).hasRank()); in TEST_F()
60 EXPECT_EQ(range.getValueAsShape(1).getRank(), 1); in TEST_F()
61 EXPECT_EQ(range.getValueAsShape(1).getDimSize(0), 10); in TEST_F()
76 ASSERT_TRUE(range.getValueAsShape(0)); in TEST_F()
77 EXPECT_TRUE(range.getValueAsShape(0).hasRank()); in TEST_F()
78 EXPECT_EQ(range.getValueAsShape(0).getRank(), 1); in TEST_F()
80 ASSERT_TRUE(range.getValueAsShape(1)); in TEST_F()
81 EXPECT_TRUE(range.getValueAsShape(1).hasRank()); in TEST_F()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h211 ShapeAdaptor getValueAsShape(int index);
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DInferTypeOpInterface.cpp149 ShapeAdaptor ValueShapeRange::getValueAsShape(int index) { in getValueAsShape() function in ValueShapeRange
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp699 if (ShapeAdaptor permShape = operands.getValueAsShape(1)) { in inferReturnTypeComponents()