| /llvm-project/mlir/test/python/dialects/linalg/opdsl/ |
| H A D | emit_pooling.py | 55 def test_f32i32_max_pooling(input, shape, init_result): argument
68 def test_f32i32_max_unsigned_pooling(input, shape, init_result): argument
92 def test_f32f32_max_pooling(input, shape, init_result): argument
105 def test_f32i32_min_pooling(input, shape, init_result): argument
123 def test_f32i32_min_unsigned_pooling(input, shape, init_result): argument
141 def test_f32f32_min_pooling(input, shape, init_result): argument
|
| /llvm-project/mlir/lib/IR/ |
| H A D | BuiltinAttributeInterfaces.cpp | 44 ArrayRef<int64_t> shape = type.getShape(); in isValidIndex() local
63 ArrayRef<int64_t> shape = shapeType.getShape(); in getFlattenedIndex() local
78 AffineMap m, ArrayRef<int64_t> shape, in verifyAffineMapAsLayout()
|
| H A D | BuiltinTypes.cpp | 229 verify(function_ref<InFlightDiagnostic ()> emitError,ArrayRef<int64_t> shape,Type elementType,ArrayRef<bool> scalableDims) verify() argument
260 cloneWith(std::optional<ArrayRef<int64_t>> shape,Type elementType) const cloneWith() argument
282 cloneWith(std::optional<ArrayRef<int64_t>> shape,Type elementType) const cloneWith() argument
298 clone(::llvm::ArrayRef<int64_t> shape,Type elementType) const clone() argument
332 verify(function_ref<InFlightDiagnostic ()> emitError,ArrayRef<int64_t> shape,Type elementType,Attribute encoding) verify() argument
369 cloneWith(std::optional<ArrayRef<int64_t>> shape,Type elementType) const cloneWith() argument
386 clone(::llvm::ArrayRef<int64_t> shape,Type elementType) const clone() argument
524 get(ArrayRef<int64_t> shape,Type elementType,MemRefLayoutAttrInterface layout,Attribute memorySpace) get() argument
540 getChecked(function_ref<InFlightDiagnostic ()> emitErrorFn,ArrayRef<int64_t> shape,Type elementType,MemRefLayoutAttrInterface layout,Attribute memorySpace) getChecked() argument
555 get(ArrayRef<int64_t> shape,Type elementType,AffineMap map,Attribute memorySpace) get() argument
575 getChecked(function_ref<InFlightDiagnostic ()> emitErrorFn,ArrayRef<int64_t> shape,Type elementType,AffineMap map,Attribute memorySpace) getChecked() argument
593 get(ArrayRef<int64_t> shape,Type elementType,AffineMap map,unsigned memorySpaceInd) get() argument
614 getChecked(function_ref<InFlightDiagnostic ()> emitErrorFn,ArrayRef<int64_t> shape,Type elementType,AffineMap map,unsigned memorySpaceInd) getChecked() argument
634 verify(function_ref<InFlightDiagnostic ()> emitError,ArrayRef<int64_t> shape,Type elementType,MemRefLayoutAttrInterface layout,Attribute memorySpace) verify() argument
[all...] |
| H A D | BuiltinTypeInterfaces.cpp | 28 getNumElements(ArrayRef<int64_t> shape) getNumElements() argument
|
| /llvm-project/llvm/test/Transforms/Inline/ML/Inputs/ |
| H A D | test_output_spec.json | 8 "shape": [ array
19 "shape": [ array
|
| /llvm-project/mlir/python/mlir/extras/ |
| H A D | types.py | 121 *shape, argument
85 _shaped(*shape, element_type: Type = None, type_constructor=None) global() argument
107 vector( *shape, element_type: Type = None, scalable: Optional[List[bool]] = None, scalable_dims: Optional[List[int]] = None, ) global() argument
138 memref( *shape, element_type: Type = None, memory_space: Optional[int] = None, layout: Optional[StridedLayoutAttr] = None, ) global() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/XeGPU/IR/ |
| H A D | XeGPUDialect.cpp | 51 llvm::SmallVector<int64_t> shape; parse() local
93 auto shape = getShape(); print() local
110 get(llvm::ArrayRef<int64_t> shape,mlir::Type elementType,bool scattered,int array_length,MemoryScope memory_scope,bool boundary_check) get() argument
[all...] |
| H A D | XeGPUOps.cpp | 45 SmallVector<int64_t> shape; in getShapeOf() local
22 transpose(llvm::ArrayRef<int64_t> trans,SmallVector<int64_t> & shape) transpose() argument
99 build(OpBuilder & builder,OperationState & state,Type tdesc,TypedValue<IntegerType> source,llvm::ArrayRef<OpFoldResult> offsets,llvm::ArrayRef<OpFoldResult> shape,llvm::ArrayRef<OpFoldResult> strides) build() argument
303 SmallVector<int64_t> shape({(int64_t)getNumOffsets()}); verify() local
[all...] |
| /llvm-project/flang/test/Lower/Intrinsics/ |
| H A D | c_f_pointer.f90 | 112 subroutine dynamic_shape_size(cptr, fptr, shape) argument
116 integer :: shape(:) local
130 subroutine dynamic_shape_size_2(cptr, fptr, shape, n) argument
135 integer :: shape(n) local
|
| H A D | reshape.f90 | 31 subroutine test_reshape_optional(pad, order, source, shape) argument
35 integer :: shape(4) local
[all...] |
| /llvm-project/flang/unittests/Runtime/ |
| H A D | Reduction.cpp | 40 std::vector<int> shape{2, 3}; in TEST() local
59 std::vector<int> shape{3, 4, 2}; // rows, columns, planes in TEST() local
254 std::vector<int> shape{2, 3}; in TEST() local
348 std::vector<int> shape{ in TEST() local
479 std::vector<int> shape{2, 3}; TEST() local
627 std::vector<int> shape{3}; TEST() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Shape/Transforms/ |
| H A D | OutlineShapeComputation.cpp | 66 Value shape, StringRef fnName, Location loc) { in createFuncFromCluster() argument
101 Value shape = it.first; getOrderedClusters() local
114 for (Value shape : it->second) getOrderedClusters() local
135 Value shape = withOp.getShape(); constructShapeFunc() local
267 Value shape = withOp.getShape(); constructClustersForEachShape() local
277 getClusterFromValue(Value shape,DenseMap<Value,DenseSet<Operation * >> & clusters) getClusterFromValue() argument
[all...] |
| /llvm-project/flang/lib/Evaluate/ |
| H A D | shape.cpp | 87 shape argument
36 const auto &shape{details->shape()}; IsExplicitShape() local
80 Shape shape; CreateShape() local
100 AsConstantShape(FoldingContext & context,const Shape & shape) AsConstantShape() argument
110 AsConstantShape(const ConstantSubscripts & shape) AsConstantShape() argument
119 AsConstantExtents(const Constant<ExtentType> & shape) AsConstantExtents() argument
128 AsConstantExtents(FoldingContext & context,const Shape & shape) AsConstantExtents() argument
136 AsShape(const ConstantSubscripts & shape) AsShape() argument
144 AsShape(const std::optional<ConstantSubscripts> & shape) AsShape() argument
152 Fold(FoldingContext & context,Shape && shape) Fold() argument
160 Fold(FoldingContext & context,std::optional<Shape> && shape) Fold() argument
198 GetSize(Shape && shape) GetSize() argument
210 GetSize(const ConstantSubscripts & shape) GetSize() argument
484 if (auto shape{GetShape(assoc->expr())}; GetAssociatedExtent() local
520 if (auto shape{GetShape(symbol, invariantOnly)}) { GetExtent() local
569 if (auto shape{GetShape(subs.value())}) { GetExtent() local
811 Shape shape; operator ()() local
833 Shape shape; operator ()() local
1081 if (auto shape{(*this)(call.arguments().at(0))}) { operator ()() local
[all...] |
| H A D | fold-matmul.h | 102 ConstantSubscripts shape; in FoldMatmul() local
|
| H A D | constant.cpp | 17 ConstantBounds::ConstantBounds(const ConstantSubscripts &shape) in ConstantBounds()
20 ConstantBounds::ConstantBounds(ConstantSubscripts &&shape) in ConstantBounds()
83 std::optional<uint64_t> TotalElementCount(const ConstantSubscripts &shape) { in TotalElementCount()
135 bool HasNegativeExtent(const ConstantSubscripts &shape) { in HasNegativeExtent()
358 std::vector<StructureConstructor> &&x, ConstantSubscripts &&shape) in Constant()
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | IRInterfaces.cpp | 322 PyShapedTypeComponents(py::list shape,MlirType elementType) PyShapedTypeComponents() argument
324 PyShapedTypeComponents(py::list shape,MlirType elementType,MlirAttribute attribute) PyShapedTypeComponents() argument
350 __anonf53964290402(py::list shape, PyType &elementType) bind() argument
357 __anonf53964290502(py::list shape, PyType &elementType, PyAttribute &attribute) bind() argument
395 py::list shape; global() member in mlir::python::PyShapedTypeComponents
421 appendResultsCallback(bool hasRank,intptr_t rank,const int64_t * shape,MlirType elementType,MlirAttribute attribute,void * userData) appendResultsCallback() argument
[all...] |
| /llvm-project/mlir/unittests/IR/ |
| H A D | ShapedTypeTest.cpp | 138 SmallVector<int64_t> shape{2, 4, 8, 9, 1}; TEST() local
176 SmallVector<int64_t> shape{1, 2, 3, 4}; TEST() local
196 SmallVector<int64_t> shape{2, 4, 8, 16, 32}; TEST() local
[all...] |
| /llvm-project/mlir/test/python/dialects/ |
| H A D | shape.py | 6 import mlir.dialects.shape as shape namespace
|
| /llvm-project/mlir/lib/CAPI/IR/ |
| H A D | BuiltinTypes.cpp | 282 mlirVectorTypeGet(intptr_t rank,const int64_t * shape,MlirType elementType) mlirVectorTypeGet() argument
289 mlirVectorTypeGetChecked(MlirLocation loc,intptr_t rank,const int64_t * shape,MlirType elementType) mlirVectorTypeGetChecked() argument
295 mlirVectorTypeGetScalable(intptr_t rank,const int64_t * shape,const bool * scalable,MlirType elementType) mlirVectorTypeGetScalable() argument
303 mlirVectorTypeGetScalableChecked(MlirLocation loc,intptr_t rank,const int64_t * shape,const bool * scalable,MlirType elementType) mlirVectorTypeGetScalableChecked() argument
344 mlirRankedTensorTypeGet(intptr_t rank,const int64_t * shape,MlirType elementType,MlirAttribute encoding) mlirRankedTensorTypeGet() argument
352 mlirRankedTensorTypeGetChecked(MlirLocation loc,intptr_t rank,const int64_t * shape,MlirType elementType,MlirAttribute encoding) mlirRankedTensorTypeGetChecked() argument
388 mlirMemRefTypeGet(MlirType elementType,intptr_t rank,const int64_t * shape,MlirAttribute layout,MlirAttribute memorySpace) mlirMemRefTypeGet() argument
399 mlirMemRefTypeGetChecked(MlirLocation loc,MlirType elementType,intptr_t rank,const int64_t * shape,MlirAttribute layout,MlirAttribute memorySpace) mlirMemRefTypeGetChecked() argument
412 mlirMemRefTypeContiguousGet(MlirType elementType,intptr_t rank,const int64_t * shape,MlirAttribute memorySpace) mlirMemRefTypeContiguousGet() argument
421 mlirMemRefTypeContiguousGetChecked(MlirLocation loc,MlirType elementType,intptr_t rank,const int64_t * shape,MlirAttribute memorySpace) mlirMemRefTypeContiguousGetChecked() argument
[all...] |
| /llvm-project/mlir/include/mlir/ExecutionEngine/ |
| H A D | MemRefUtils.h | 46 inline std::array<int64_t, N> makeStrides(ArrayRef<int64_t> shape) { in makeStrides()
65 makeStridedMemRefDescriptor(T *ptr, T *alignedPtr, ArrayRef<int64_t> shape, in makeStridedMemRefDescriptor()
|
| /llvm-project/mlir/test/python/integration/dialects/linalg/ |
| H A D | opsrun.py | 519 pooling_on_buffers(input, shape, output) global() argument
553 pooling_on_buffers(input, shape, output) global() argument
593 pooling_on_buffers(input, shape, output) global() argument
625 pooling_on_buffers(input, shape, output) global() argument
[all...] |
| /llvm-project/flang/lib/Optimizer/Builder/ |
| H A D | HLFIRTools.cpp | 28 getExplicitExtentsFromShape(mlir::Value shape,fir::FirOpBuilder & builder) getExplicitExtentsFromShape() argument
66 if (mlir::Value shape = var.getShape()) getExplicitExtents() local
74 getExplicitLboundsFromShape(mlir::Value shape) getExplicitLboundsFromShape() argument
92 if (mlir::Value shape = var.getShape()) getExplicitLbounds() local
200 isShapeWithLowerBounds(mlir::Value shape) isShapeWithLowerBounds() argument
267 mlir::Value shape{}; genAssociateExpr() local
442 genBounds(mlir::Location loc,fir::FirOpBuilder & builder,mlir::Value shape) genBounds() argument
464 genLowerbounds(mlir::Location loc,fir::FirOpBuilder & builder,mlir::Value shape,unsigned rank) genLowerbounds() argument
549 if (auto shape = tryRetrievingShapeOrShift(entity)) { genShape() local
566 getIndexExtents(mlir::Location loc,fir::FirOpBuilder & builder,mlir::Value shape) getIndexExtents() argument
578 if (auto shape = tryRetrievingShapeOrShift(entity)) { genExtent() local
601 if (auto shape = tryRetrievingShapeOrShift(entity)) { genLBound() local
682 asEmboxShape(mlir::Location loc,fir::FirOpBuilder & builder,const fir::ExtendedValue & exv,mlir::Value shape) asEmboxShape() argument
771 getArrayExprType(mlir::Type elementType,mlir::Value shape,bool isPolymorphic) getArrayExprType() argument
784 genElementalOp(mlir::Location loc,fir::FirOpBuilder & builder,mlir::Type elementType,mlir::Value shape,mlir::ValueRange typeParams,const ElementalKernelGenerator & genKernel,bool isUnordered,mlir::Value polymorphicMold,mlir::Type exprType) genElementalOp() argument
1103 mlir::Value shape{}; createTempFromMold() local
1164 mlir::Value shape{}; createStackTempFromMold() local
1238 mlir::Value shape = hlfir::genShape(loc, builder, source); genTypeAndKindConvert() local
[all...] |
| /llvm-project/flang/lib/Optimizer/HLFIR/IR/ |
| H A D | HLFIRDialect.cpp | 50 ExprType::Shape shape; in parse() local
67 auto shape = getShape(); in print() local
210 fir::ShapeOp shape = builder.create<fir::ShapeOp>(loc, shapeTy, extents); in genExprShape() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | TransposeMatmul.cpp | 48 ArrayRef<int64_t> shape = type.getShape(); transposeMatmul() local
100 ArrayRef<int64_t> shape = type.getShape(); transposeBatchMatmul() local
|
| /llvm-project/mlir/lib/Dialect/Math/Transforms/ |
| H A D | PolynomialApproximation.cpp | 68 broadcast(Type type,VectorShape shape) broadcast() argument
77 broadcast(ImplicitLocOpBuilder & builder,Value value,VectorShape shape) broadcast() argument
230 VectorShape shape = vectorShape(arg); frexp() local
270 VectorShape shape = vectorShape(arg); exp2I32() local
296 VectorShape shape = vectorShape(x); makePolynomialCalculation() local
394 VectorShape shape = vectorShape(op.getOperand()); matchAndRewrite() local
493 VectorShape shape = vectorShape(op.getResult()); matchAndRewrite() local
559 VectorShape shape = vectorShape(op.getOperand()); matchAndRewrite() local
647 VectorShape shape = vectorShape(op.getOperand()); logMatchAndRewrite() local
794 VectorShape shape = vectorShape(op.getOperand()); matchAndRewrite() local
849 VectorShape shape = vectorShape(operand); matchAndRewrite() local
913 VectorShape shape = vectorShape(operand); matchAndRewrite() local
991 VectorShape shape = vectorShape(operand); matchAndRewrite() local
1100 clampWithNormals(ImplicitLocOpBuilder & builder,const VectorShape shape,Value value,float lowerBound,float upperBound) clampWithNormals() argument
1135 auto shape = vectorShape(op.getOperand().getType()); matchAndRewrite() local
1292 VectorShape shape = vectorShape(op.getOperand()); matchAndRewrite() local
1362 VectorShape shape = vectorShape(op.getOperand()); matchAndRewrite() local
1489 VectorShape shape = vectorShape(operand); matchAndRewrite() local
1578 VectorShape shape = vectorShape(op.getOperand()); matchAndRewrite() local
[all...] |