|
SmallVector< utils::IteratorType > | mlir::tosa::getNParallelLoopsAttrs (unsigned nParallelLoops) |
|
SmallVector< Value > | mlir::tosa::condenseValues (const SmallVector< Value > &values) |
|
Value | mlir::tosa::clampFloatHelper (Location loc, Value arg, Value min, Value max, OpBuilder &rewriter) |
|
Value | mlir::tosa::clampIntHelper (Location loc, Value arg, Value min, Value max, OpBuilder &rewriter, bool isUnsigned) |
|
bool | mlir::tosa::validIntegerRange (IntegerType ty, int64_t value) |
|
template<typename Op > |
std::optional< SmallVector< Value > > | mlir::tosa::checkHasDynamicBatchDims (PatternRewriter &rewriter, Op op, ArrayRef< Value > params) |
|
LogicalResult | mlir::tosa::EqualizeRanks (PatternRewriter &rewriter, Location loc, Value &input1, Value &input2) |
| Common code to create the reshape op where necessary to make the rank of two values equal. More...
|
|
LogicalResult | mlir::tosa::EqualizeRanks (ImplicitLocOpBuilder &builder, Value &input1, Value &input2) |
|
template<typename TosaOp , typename... Args> |
TosaOp | mlir::tosa::CreateOpAndInferShape (ImplicitLocOpBuilder &builder, Type resultTy, Args &&...args) |
|
template<typename TosaOp , typename... Args> |
TosaOp | mlir::tosa::CreateOpAndInferShape (PatternRewriter &rewriter, Location loc, Type resultTy, Args &&...args) |
|
template<typename T > |
SmallVector< T > | mlir::tosa::applyTOSAPermutation (ArrayRef< T > input, ArrayRef< int32_t > perms) |
|