Home
last modified time | relevance | path

Searched refs:ShapedTypeComponents (Results 1 – 9 of 9) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h28 class ShapedTypeComponents; variable
43 ShapeAdaptor(ShapedTypeComponents *components) : val(components) {} in ShapeAdaptor()
44 ShapeAdaptor(ShapedTypeComponents &components) : val(&components) {} in ShapeAdaptor()
58 void getDims(ShapedTypeComponents &res) const;
101 class ShapedTypeComponents {
107 ShapedTypeComponents() : elementType(nullptr), attr(nullptr){}; in ShapedTypeComponents() function
108 ShapedTypeComponents(Type elementType) in ShapedTypeComponents() function
110 ShapedTypeComponents(ShapedType shapedType) : attr(nullptr) { in ShapedTypeComponents() function
116 ShapedTypeComponents(ShapeAdaptor adaptor) : attr(nullptr) { in ShapedTypeComponents() function
124 ShapedTypeComponents(Arg &&arg, Type elementType = nullptr,
[all …]
H A DInferTypeOpInterface.td151 "::llvm::SmallVectorImpl<::mlir::ShapedTypeComponents>&":
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp351 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
363 inferredReturnShapes.push_back(ShapedTypeComponents(outShape)); in inferReturnTypeComponents()
399 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
450 inferredReturnShapes.push_back(ShapedTypeComponents(outShape)); in inferReturnTypeComponents()
479 inferredReturnShapes.push_back(ShapedTypeComponents(outShape)); in inferReturnTypeComponents()
494 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
502 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
561 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
654 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
842 inferredReturnShapes.push_back(ShapedTypeComponents()); in ReduceInferReturnTypes()
[all …]
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DInferTypeOpInterface.cpp32 return val.get<ShapedTypeComponents *>()->hasRank(); in hasRank()
42 return val.get<ShapedTypeComponents *>()->getElementType(); in getElementType()
57 auto vals = val.get<ShapedTypeComponents *>()->getDims(); in getDims()
62 void ShapeAdaptor::getDims(ShapedTypeComponents &res) const { in getDims()
76 auto *stc = val.get<ShapedTypeComponents *>(); in getDimSize()
86 return val.get<ShapedTypeComponents *>()->getDims().size(); in getRank()
102 auto *stc = val.get<ShapedTypeComponents *>(); in hasStaticShape()
122 auto *stc = val.get<ShapedTypeComponents *>(); in getNumElements()
180 SmallVectorImpl<ShapedTypeComponents> &retComponents)> in inferReturnTensorTypes()
185 SmallVector<ShapedTypeComponents, 2> retComponents; in inferReturnTensorTypes()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp37 Operation &op, DenseMap<Value, ShapedTypeComponents> &shapesStorage) { in propagateShapesToTosaIf()
75 Operation &op, DenseMap<Value, ShapedTypeComponents> &shapesStorage) { in propagateShapesToTosaWhile()
182 DenseMap<Value, ShapedTypeComponents> shapesStorage; in propagateShapesInRegion()
210 SmallVector<ShapedTypeComponents> returnedShapes; in propagateShapesInRegion()
220 ShapedTypeComponents predictedShape = std::get<1>(it); in propagateShapesInRegion()
H A DTosaDecomposeTransposeConv.cpp46 SmallVector<ShapedTypeComponents> returnedShapes; in createOpAndInfer()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h65 ShapedTypeComponents getShapedTypeComponents() const { in getShapedTypeComponents()
66 return hasRank ? ShapedTypeComponents(sizes) : ShapedTypeComponents(); in getShapedTypeComponents()
/llvm-project-15.0.7/mlir/unittests/Interfaces/
H A DInferTypeOpInterfaceTest.cpp68 ShapedTypeComponents fixed(SmallVector<int64_t>{30}); in TEST_F()
87 ShapedTypeComponents shape(SmallVector<int64_t>{10, 20}); in TEST_F()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Test/
H A DTestDialect.cpp1164 SmallVectorImpl<ShapedTypeComponents> &inferredReturnShapes) { in inferReturnTypeComponents()
1174 inferredReturnShapes.push_back(ShapedTypeComponents({dim}, type)); in inferReturnTypeComponents()