Searched refs:shape2 (Results 1 – 7 of 7) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/ |
| H A D | Traits.cpp | 17 ArrayRef<int64_t> shape2) { in staticallyKnownBroadcastable() argument
20 extents.emplace_back(shape2.begin(), shape2.end()); in staticallyKnownBroadcastable()
60 ArrayRef<int64_t> shape2, in getBroadcastedShape() argument
71 if (shape1.size() > shape2.size()) { in getBroadcastedShape()
74 std::copy(shape2.begin(), shape2.end(), std::back_inserter(resultShape)); in getBroadcastedShape()
78 auto i2 = shape2.rbegin(), e2 = shape2.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,
64 ArrayRef<int64_t> shape2);
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | TypeUtilities.cpp | 58 ArrayRef<int64_t> shape2) { 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/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 217 auto shape2 = tp2.getShape(); in verify() local
222 if (shape1[d] != shape2[d] && shape2[d] != ShapedType::kDynamicSize) in verify()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | TypeUtilities.h | 49 ArrayRef<int64_t> shape2);
|
| /llvm-project-15.0.7/mlir/test/Dialect/Tensor/ |
| H A D | ops.mlir | 77 %shape2: tensor<2xi32>, %shape3: tensor<?xi32>) -> tensor<*xf32> {
80 %dyn_mat = tensor.reshape %dyn_vec(%shape2)
|
| /llvm-project-15.0.7/mlir/test/Dialect/MemRef/ |
| H A D | ops.mlir | 41 %shape2: memref<2xi32>, %shape3: memref<?xi32>) -> memref<*xf32> {
44 %dyn_mat = memref.reshape %dyn_vec(%shape2)
|