Lines Matching full:matmul
40 %matmul = linalg.matmul
46 ins(%matmul, %bias : tensor<512x512xf32>, tensor<512x512xf32>)
61 additional arguments, `bind-first-extra-to-ops=linalg.matmul
84 %matmul = transform.collect_matching @match_matmul in %root
88 transform.include @print_matmul failures(propagate) (%matmul)
106 transform.match.operation_name %entry ["linalg.matmul"] : !transform.any_op
118 %matmul: !transform.any_op {transform.readonly}) {
119 transform.debug.emit_remark_at %matmul, "matmul" : !transform.any_op
130 `linalg.elemwise_binary` and `linalg.matmul` operations. In debug builds, the
139 [transform-matcher] matching %0 = linalg.matmul ins(%arg0, %arg1 : tensor<512x512xf32>, tensor<512x…
146 substituting `%arg1` with `%matmul` and `%arg2` with `%elemwise`.
156 specific group of operations where a `matmul` operation produces a result that
186 %matmul = transform.get_producer_of_operand %middle[0]
188 // And that operation is a matmul.
189 transform.match.operation_name %matmul ["linalg.matmul"] : !transform.any_op
190 // We will yield the handles to the matmul and the two elementwise
192 transform.yield %matmul, %middle, %last
197 This matcher is applicable in presence of other `elemwise` and `matmul`
208 %matmul, %el1, %el2 = transform.collect_matching @match_matmul_elemwise in %root
214 transform.include @print_matmul failures(propagate) (%matmul)
392 the matmul to produce the first operand for simplicity. The updated matcher
423 %matmul = transform.get_producer_of_operand %middle[0]
425 // And that operation is a matmul.
426 transform.match.operation_name %matmul ["linalg.matmul"] : !transform.any_op
427 // We will yield the handles to the matmul and the two elementwise
429 transform.yield %matmul, %middle, %last, %pos
436 This achieves the desired effect and matches both `max(add(matmul(...), bias),
437 0)` and `max(0, add(matmul(...), bias))` in the same values. The `%pos` value is
483 tensor contraction using `linalg.generic` instead of `linalg.matmul`, the