Searched refs:PoolingOp (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgOps.h | 95 template <typename PoolingOp>
97 weightedPoolingInputIndex(PoolingOp op, ArrayRef<AffineExpr> outputDims,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 41 class PoolingOp : public XlaOpKernel { class
43 PoolingOp(OpKernelConstruction* ctx, int num_spatial_dims, in PoolingOp() function in tensorflow::__anon9297e9860111::PoolingOp
152 class MaxPoolOp : public PoolingOp {
155 : PoolingOp(ctx, /*num_spatial_dims=*/num_spatial_dims, in MaxPoolOp()
210 class AvgPoolOp : public PoolingOp {
213 : PoolingOp(ctx, /*num_spatial_dims=*/num_spatial_dims, in AvgPoolOp()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Loops.cpp | 370 template <typename PoolingOp>
371 static bool hasPadding(PoolingOp poolingOp) { in hasPadding()
379 template <typename IndexedValueType, typename PoolingOp>
380 static Value getPoolingInput(PoolingOp op, ArrayRef<Value> inputIndices) { in getPoolingInput()
384 Value padValue = std_constant(type, getPadValueAttr<PoolingOp>(type)); in getPoolingInput()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 1259 template <typename PoolingOp>
1260 static LogicalResult verifySingleInputPoolingOp(PoolingOp op) { in verifySingleInputPoolingOp()
1360 template <typename PoolingOp>
1362 mlir::linalg::weightedPoolingInputIndex(PoolingOp op, in weightedPoolingInputIndex()
|