Home
last modified time | relevance | path

Searched defs:shape (Results 1 – 25 of 141) sorted by relevance

123456

/llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/
H A Demit_pooling.py53 def test_f32i32_max_pooling(input, shape, init_result): argument
64 def test_f32i32_max_unsigned_pooling(input, shape, init_result): argument
86 def test_f32f32_max_pooling(input, shape, init_result): argument
97 def test_f32i32_min_pooling(input, shape, init_result): argument
113 def test_f32i32_min_unsigned_pooling(input, shape, init_result): argument
129 def test_f32f32_min_pooling(input, shape, init_result): argument
/llvm-project-15.0.7/mlir/lib/IR/
H A DBuiltinAttributeInterfaces.cpp48 ArrayRef<int64_t> shape = type.getShape(); in isValidIndex() local
67 ArrayRef<int64_t> shape = shapeType.getShape(); in getFlattenedIndex() local
82 AffineMap m, ArrayRef<int64_t> shape, in verifyAffineMapAsLayout()
H A DBuiltinTypes.cpp233 ArrayRef<int64_t> shape, Type elementType, in verify()
271 VectorType VectorType::cloneWith(Optional<ArrayRef<int64_t>> shape, in cloneWith()
293 TensorType TensorType::cloneWith(Optional<ArrayRef<int64_t>> shape, in cloneWith()
334 ArrayRef<int64_t> shape, Type elementType, in verify()
396 BaseMemRefType BaseMemRefType::cloneWith(Optional<ArrayRef<int64_t>> shape, in cloneWith()
549 MemRefType MemRefType::get(ArrayRef<int64_t> shape, Type elementType, in get()
565 function_ref<InFlightDiagnostic()> emitErrorFn, ArrayRef<int64_t> shape, in getChecked()
580 MemRefType MemRefType::get(ArrayRef<int64_t> shape, Type elementType, in get()
618 MemRefType MemRefType::get(ArrayRef<int64_t> shape, Type elementType, in get()
659 ArrayRef<int64_t> shape, Type elementType, in verify()
[all …]
H A DTypeUtilities.cpp151 [&](auto shape) { return firstRank != shape.getRank(); })) in verifyCompatibleShapes()
158 shapes, [&](auto shape) { return shape.getRank() >= i; }), in verifyCompatibleShapes()
159 [&](auto shape) { return shape.getDimSize(i); })); in verifyCompatibleShapes()
H A DBuiltinTypeInterfaces.cpp29 int64_t ShapedType::getNumElements(ArrayRef<int64_t> shape) { in getNumElements()
/llvm-project-15.0.7/flang/lib/Evaluate/
H A Dshape.cpp36 const auto &shape{details->shape()}; in IsExplicitShape() local
80 Shape shape; in CreateShape() local
100 FoldingContext &context, const Shape &shape) { in AsConstantShape()
128 FoldingContext &context, const Shape &shape) { in AsConstantExtents()
136 Shape AsShape(const ConstantSubscripts &shape) { in AsShape()
198 MaybeExtentExpr GetSize(Shape &&shape) { in GetSize()
450 if (auto shape{GetShape(symbol)}) { in GetExtent() local
550 if (auto shape{GetShape(assoc->expr())}) { in GetRawUpperBound() local
600 if (auto shape{GetShape(assoc->expr())}) { in GetUBOUND() local
730 Shape shape; in operator ()() local
[all …]
H A Dconstant.cpp17 ConstantBounds::ConstantBounds(const ConstantSubscripts &shape) in ConstantBounds()
20 ConstantBounds::ConstantBounds(ConstantSubscripts &&shape) in ConstantBounds()
74 std::size_t TotalElementCount(const ConstantSubscripts &shape) { in TotalElementCount()
116 bool HasNegativeExtent(const ConstantSubscripts &shape) { in HasNegativeExtent()
332 std::vector<StructureConstructor> &&x, ConstantSubscripts &&shape) in Constant()
H A Dfold-reduction.cpp34 std::optional<ActualArgument> &maskArg, const ConstantSubscripts &shape, in GetReductionMASK()
H A Dfold-implementation.h469 ConstantSubscripts shape; in FoldElementalIntrinsicHelper() local
1314 ArrayConstructor<T> &&values, std::optional<ConstantSubscripts> &&shape) { in FromArrayConstructor()
1334 std::function<Expr<RESULT>(Expr<OPERAND> &&)> &&f, const Shape &shape, in MapOperation()
1374 const Shape &shape, std::optional<Expr<SubscriptInteger>> &&length, in MapOperation()
1415 const Shape &shape, std::optional<Expr<SubscriptInteger>> &&length, in MapOperation()
1432 const Shape &shape, std::optional<Expr<SubscriptInteger>> &&length, in MapOperation()
/llvm-project-15.0.7/llvm/test/Transforms/Inline/ML/Inputs/
H A Dtest_output_spec.json8 "shape": [ array
19 "shape": [ array
/llvm-project-15.0.7/flang/unittests/Runtime/
H A DReduction.cpp38 std::vector<int> shape{2, 3}; in TEST() local
57 std::vector<int> shape{3, 4, 2}; // rows, columns, planes in TEST() local
252 std::vector<int> shape{2, 3}; in TEST() local
346 std::vector<int> shape{2, 2}; in TEST() local
477 std::vector<int> shape{2, 3}; in TEST() local
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DBuiltinTypes.cpp172 MlirType mlirVectorTypeGet(intptr_t rank, const int64_t *shape, in mlirVectorTypeGet()
180 const int64_t *shape, MlirType elementType) { in mlirVectorTypeGetChecked()
200 MlirType mlirRankedTensorTypeGet(intptr_t rank, const int64_t *shape, in mlirRankedTensorTypeGet()
208 const int64_t *shape, in mlirRankedTensorTypeGetChecked()
236 const int64_t *shape, MlirAttribute layout, in mlirMemRefTypeGet()
247 intptr_t rank, const int64_t *shape, in mlirMemRefTypeGetChecked()
260 const int64_t *shape, in mlirMemRefTypeContiguousGet()
269 const int64_t *shape, in mlirMemRefTypeContiguousGetChecked()
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DVectorInterfaces.cpp16 SmallVector<int64_t, 8> shape; in transferMaskType() local
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypes.h173 Builder(ArrayRef<int64_t> shape, Type elementType) in Builder()
224 Builder(ArrayRef<int64_t> shape, Type elementType, Attribute encoding) in Builder()
290 : shape(shape), elementType(elementType), in shape() argument
/llvm-project-15.0.7/mlir/test/python/dialects/
H A Dshape.py6 import mlir.dialects.shape as shape namespace
/llvm-project-15.0.7/flang/test/Lower/Intrinsics/
H A Dreshape.f9031 subroutine test_reshape_optional(pad, order, source, shape) argument
35 integer :: shape(4) local
/llvm-project-15.0.7/mlir/test/python/integration/dialects/linalg/
H A Dopsrun.py492 def pooling_on_buffers(input, shape, output): argument
523 def pooling_on_buffers(input, shape, output): argument
560 def pooling_on_buffers(input, shape, output): argument
590 def pooling_on_buffers(input, shape, output): argument
/llvm-project-15.0.7/mlir/include/mlir/ExecutionEngine/
H A DMemRefUtils.h45 inline std::array<int64_t, N> makeStrides(ArrayRef<int64_t> shape) { in makeStrides()
64 makeStridedMemRefDescriptor(T *ptr, T *alignedPtr, ArrayRef<int64_t> shape, in makeStridedMemRefDescriptor()
/llvm-project-15.0.7/mlir/unittests/IR/
H A DAttributeTest.cpp18 RankedTensorType shape = RankedTensorType::get({2, 1}, eltType); in testSplat() local
36 RankedTensorType shape = RankedTensorType::get({2, 2}, boolTy); in TEST() local
61 RankedTensorType shape = RankedTensorType::get({boolCount}, boolTy); in TEST() local
84 RankedTensorType shape = RankedTensorType::get({6}, boolTy); in TEST() local
201 RankedTensorType shape = RankedTensorType::get({}, intTy); in TEST() local
/llvm-project-15.0.7/flang/test/Semantics/
H A Dselecttype02.f904 type shape type
/llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/
H A DPolynomialApproximation.cpp56 static Type broadcast(Type type, ArrayRef<int64_t> shape) { in broadcast()
63 ArrayRef<int64_t> shape) { in broadcast()
210 ArrayRef<int64_t> shape = vectorShape(arg); in frexp() local
250 ArrayRef<int64_t> shape = vectorShape(arg); in exp2I32() local
274 ArrayRef<int64_t> shape = vectorShape(x); in makePolynomialCalculation() local
371 ArrayRef<int64_t> shape = vectorShape(op.getOperand()); in matchAndRewrite() local
429 ArrayRef<int64_t> shape = vectorShape(op.getResult()); in matchAndRewrite() local
495 ArrayRef<int64_t> shape = vectorShape(op.getOperand()); in matchAndRewrite() local
583 ArrayRef<int64_t> shape = vectorShape(op.getOperand()); in logMatchAndRewrite() local
730 ArrayRef<int64_t> shape = vectorShape(op.getOperand()); in matchAndRewrite() local
[all …]
/llvm-project-15.0.7/flang/unittests/Evaluate/
H A Dreshape.cpp33 auto shape{Descriptor::Create(TypeCategory::Integer, in main() local
/llvm-project-15.0.7/mlir/unittests/ExecutionEngine/
H A DInvoke.cpp147 int64_t shape[] = {9}; in TEST() local
194 int64_t shape[] = {k, m}; in TEST() local
251 int64_t shape[] = {k, m}; in TEST() local
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusePadOpWithLinalgProducer.cpp96 for (const auto &shape : llvm::enumerate( in matchAndRewrite() local
/llvm-project-15.0.7/mlir/lib/Dialect/Quant/Utils/
H A DUniformSupport.cpp78 auto shape = type.getShape(); in convert() local

123456