Searched refs:shape1 (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/ |
| H A D | Traits.cpp | 16 bool OpTrait::util::staticallyKnownBroadcastable(ArrayRef<int64_t> shape1, in staticallyKnownBroadcastable() argument
19 extents.emplace_back(shape1.begin(), shape1.end()); in staticallyKnownBroadcastable()
59 bool OpTrait::util::getBroadcastedShape(ArrayRef<int64_t> shape1, in getBroadcastedShape() argument
71 if (shape1.size() > shape2.size()) { in getBroadcastedShape()
72 std::copy(shape1.begin(), shape1.end(), std::back_inserter(resultShape)); in getBroadcastedShape()
77 auto i1 = shape1.rbegin(), e1 = shape1.rend(); in getBroadcastedShape()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/ |
| H A D | Traits.h | 47 bool getBroadcastedShape(ArrayRef<int64_t> shape1, ArrayRef<int64_t> shape2,
63 bool staticallyKnownBroadcastable(ArrayRef<int64_t> shape1,
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | TypeUtilities.cpp | 57 LogicalResult mlir::verifyCompatibleShape(ArrayRef<int64_t> shape1, in verifyCompatibleShape() argument
59 if (shape1.size() != shape2.size()) in verifyCompatibleShape()
61 for (auto dims : llvm::zip(shape1, shape2)) { in verifyCompatibleShape()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | TypeUtilities.h | 48 LogicalResult verifyCompatibleShape(ArrayRef<int64_t> shape1,
|
| /llvm-project-15.0.7/flang/unittests/Runtime/ |
| H A D | Reduction.cpp | 151 std::vector<int> shape1{24}; in TEST() local
152 auto array1{MakeArray<TypeCategory::Real, 8>(shape1, rawData)}; in TEST()
374 std::vector<int> shape1{4}; in TEST() local
376 shape1, std::vector<std::int32_t>{false, false, true, true})}; in TEST()
571 std::vector<int> shape1{6}; in TEST() local
572 auto realArray1{MakeArray<TypeCategory::Real, 8>(shape1, in TEST()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 216 auto shape1 = tp1.getShape(); in verify() local
222 if (shape1[d] != shape2[d] && shape2[d] != ShapedType::kDynamicSize) in verify()
|
| /llvm-project-15.0.7/mlir/test/Dialect/Tensor/ |
| H A D | ops.mlir | 76 func.func @tensor_reshape(%unranked: tensor<*xf32>, %shape1: tensor<1xi32>,
78 %dyn_vec = tensor.reshape %unranked(%shape1)
|
| /llvm-project-15.0.7/mlir/test/Dialect/MemRef/ |
| H A D | ops.mlir | 40 func.func @memref_reshape(%unranked: memref<*xf32>, %shape1: memref<1xi32>,
42 %dyn_vec = memref.reshape %unranked(%shape1)
|