Home
last modified time | relevance | path

Searched refs:getReductionIteratorTypeName (Results 1 – 7 of 7) sorted by relevance

/external/llvm-project/mlir/include/mlir/Dialect/Utils/
DStructuredOpsUtils.h85 constexpr StringRef getReductionIteratorTypeName() { return "reduction"; } in getReductionIteratorTypeName() function
88 return strAttr && strAttr.getValue() == getReductionIteratorTypeName(); in isReductionIterator()
101 getReductionIteratorTypeName(), in getAllIteratorTypeNames()
131 return getReductionIteratorTypeName(); in toString()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
DLinalgStructuredOpsInterface.td60 return getNumIterators(getReductionIteratorTypeName(),
73 return getDimsOfType($_op, getReductionIteratorTypeName(), res);
125 getNumIterators(getReductionIteratorTypeName(), iters) == 1;
DLinalgStructuredOps.td342 iters.append(nRed, getReductionIteratorTypeName());
/external/llvm-project/mlir/lib/Dialect/Linalg/Utils/
DUtils.cpp69 return strAttr.getValue() == getReductionIteratorTypeName(); in isReductionIteratorType()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc59 res.append(nReduction, getReductionIteratorTypeName()); in GetParallelAndReductionIterators()
950 types.push_back(is_reduced ? getReductionIteratorTypeName() in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorOps.cpp481 if (iteratorTypeName == getReductionIteratorTypeName()) { in getIterationBounds()
513 getReductionIteratorTypeName(), getContext()); in getContractingDimMap()
/external/llvm-project/mlir/docs/Dialects/
DLinalg.md658 getReductionIteratorTypeName() };