Home
last modified time | relevance | path

Searched refs:ShapeAdaptor (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DInferTypeOpInterface.cpp25 bool ShapeAdaptor::hasRank() const { in hasRank()
35 Type ShapeAdaptor::getElementType() const { in getElementType()
68 int64_t ShapeAdaptor::getDimSize(int index) const { in getDimSize()
80 int64_t ShapeAdaptor::getRank() const { in getRank()
89 bool ShapeAdaptor::hasStaticShape() const { in hasStaticShape()
106 int64_t ShapeAdaptor::getNumElements() const { in getNumElements()
131 void ShapeAdaptor::dump() const { in dump()
152 if (ShapeAdaptor ret = valueToShape(val)) in getValueAsShape()
163 ShapeAdaptor ValueShapeRange::getShape(Value val) const { in getShape()
165 if (ShapeAdaptor ret = operandShape(val)) in getShape()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h33 class ShapeAdaptor {
35 ShapeAdaptor(Type t) { in ShapeAdaptor() function
39 ShapeAdaptor(Attribute t) { in ShapeAdaptor() function
43 ShapeAdaptor(ShapedTypeComponents *components) : val(components) {} in ShapeAdaptor() function
44 ShapeAdaptor(ShapedTypeComponents &components) : val(&components) {} in ShapeAdaptor() function
116 ShapedTypeComponents(ShapeAdaptor adaptor) : attr(nullptr) { in ShapedTypeComponents()
150 friend class ShapeAdaptor;
166 using ValueShapeMapFn = function_ref<ShapeAdaptor(Value)>;
211 ShapeAdaptor getValueAsShape(int index);
216 ShapeAdaptor getShape(int index) const;
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp346 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
428 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
430 ShapeAdaptor biasShape = operands.getShape(2); in inferReturnTypeComponents()
460 ShapeAdaptor lhsShape = operands.getShape(0); in inferReturnTypeComponents()
461 ShapeAdaptor rhsShape = operands.getShape(1); in inferReturnTypeComponents()
487 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
558 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
576 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
609 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
647 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
[all …]
/llvm-project-15.0.7/mlir/unittests/Interfaces/
H A DInferTypeOpInterfaceTest.cpp69 auto mapping = [&](Value val) -> ShapeAdaptor { in TEST_F()
89 auto mapping = [&](Value val) -> ShapeAdaptor { in TEST_F()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp189 auto operandShape = [&](Value val) -> ShapeAdaptor { in propagateShapesInRegion()