Lines Matching +full:batch +full:- +full:reduce

1 //===- LinalgInterface.h - Linalg operations interfaces -------------------===//
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7 //===----------------------------------------------------------------------===//
11 //===----------------------------------------------------------------------===//
44 SmallVector<unsigned, 2> batch;
53 /// 1. The m dimension is involved in an outer-product along LHS
55 /// 2. The n dimension is involved in an outer-product along RHS
59 /// 5. Optional batch dimensions that appear in all operands are captured.
62 /// When multiple dimension occurrences exist that match `batch`, `m`, `n`, or
76 SmallVector<unsigned, 2> batch;
91 /// 1. Optional batch dimensions that appear in the input and filter.
92 /// 2. The output_image dimension is involved in a cross-correlation along LHS
95 /// 3. Optional output_channel dimension is involved in an outer-product along
101 /// represents the shape of the kernel cross-correlated along a
113 /// By default, we require the `linalgOp` to have non-empty convolved dims
114 /// (implicitly non-empty `output_image` and `filter_loop`).
152 /// %0 = <elemwise>(permutation-of(cu(block-argument-0),
153 /// cu(block-argument-1)))
154 /// %1 = <reduce>(permutation-of(cu(%0), cu(block-argument-2)))
155 /// return-like cu(%1)
157 /// where <elemwise> and <reduce> are binary operations constituting a
159 /// <reduce> is an addition). The name and other properties of these operations
161 /// through a chain of side effect-free unary operations, such as casts, which