Home
last modified time | relevance | path

Searched refs:broadcastable (Results 1 – 12 of 12) sorted by relevance

/llvm-project-15.0.7/mlir/test/Dialect/
H A Dtraits.mlir3 // Verify that ops with broadcastable trait verifies operand and result type
8 %0 = "test.broadcastable"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<i32>
16 %0 = "test.broadcastable"(%arg0, %arg1) : (tensor<4xi32>, tensor<i32>) -> tensor<4xi32>
25 …%0 = "test.broadcastable"(%arg0, %arg1) : (tensor<4x3x2xi32>, tensor<3x1xi32>) -> tensor<4x3x2xi32>
62 %0 = "test.broadcastable"(%arg0, %arg1) : (tensor<4xf32>, tensor<4xf32>) -> vector<4xf32>
100 %0 = "test.broadcastable"(%arg0, %arg1) : (tensor<2xi32>, tensor<2xi32>) -> tensor<*xi32>
108 %0 = "test.broadcastable"(%arg0, %arg1) : (tensor<4x3x2xi32>, tensor<?xi32>) -> tensor<4x3x2xi32>
115 %0 = "test.broadcastable"(%arg0, %arg1) : (tensor<?x6x1xi32>, tensor<*xi32>) -> tensor<?x6x6xi32>
124 %0 = "test.broadcastable"(%arg0, %arg1) : (tensor<*xi32>, tensor<*xi32>) -> tensor<2xi32>
142 %0 = "test.broadcastable"(%arg0, %arg1) : (tensor<3x2xi32>, tensor<*xi32>) -> tensor<2xi32>
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/ShapeToStandard/
H A DShapeToStandard.td20 $_builder.getStringAttr("required broadcastable shapes")
H A DShapeToStandard.cpp283 Value broadcastable = iterArgs[0]; in matchAndRewrite() local
289 broadcastable = in matchAndRewrite()
294 b.create<scf::YieldOp>(loc, broadcastable); in matchAndRewrite()
310 loc, broadcastable, in matchAndRewrite()
318 b.create<scf::YieldOp>(loc, broadcastable); in matchAndRewrite()
/llvm-project-15.0.7/mlir/test/Conversion/ShapeToStandard/
H A Dconvert-shape-constraints.mlir9 // CHECK: cf.assert %[[BROADCAST_IS_VALID]], "required broadcastable shapes"
H A Dshape-to-standard.mlir533 // CHECK: %[[RESULT:.*]] = shape.cstr_require %[[ALL_RESULT]], "required broadcastable sh…
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Transforms/
H A DPasses.td42 def TosaMakeBroadcastable : Pass<"tosa-make-broadcastable", "func::FuncOp"> {
/llvm-project-15.0.7/mlir/test/Dialect/Shape/
H A Dcanonicalize.mlir713 // Broadcastable with broadcastable constant shapes can be removed.
727 // Empty shape arguments can be removed from broadcastable ops.
741 // Broadcastable with non-broadcastable constant shapes is always false
757 // Broadcastable without guaranteed broadcastable shapes cannot be removed.
796 // Fold ternary broadcastable
811 // Fold ternary broadcastable with dynamic ranks
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Shape/IR/
H A DShapeOps.td259 if they are broadcastable. This broadcastable follows the same logic as what
898 are broadcastable. This broadcastable follows the same logic as what
/llvm-project-15.0.7/mlir/test/Dialect/Tosa/
H A Dbroadcast.mlir1 // RUN: mlir-opt --tosa-make-broadcastable %s | FileCheck %s
/llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/
H A DShape.cpp494 auto broadcastable = operand.getDefiningOp<CstrBroadcastableOp>(); in matchAndRewrite() local
495 if (!broadcastable) in matchAndRewrite()
498 operands.insert(broadcastable); in matchAndRewrite()
/llvm-project-15.0.7/mlir/docs/
H A DShapeInference.md246 `SameOperandAndResultType` or broadcastable) we should generate the shape
/llvm-project-15.0.7/mlir/test/lib/Dialect/Test/
H A DTestOps.td615 def BroadcastableOp : TEST_Op<"broadcastable", [ResultsBroadcastableShape]> {