MLIR  20.0.0git
Namespaces | Functions
Utils.h File Reference
#include "mlir/Dialect/Tensor/IR/Tensor.h"

Go to the source code of this file.

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::tensor
 

Functions

PadOp mlir::tensor::createPadHighOp (RankedTensorType type, Value source, Value pad, bool nofold, Location loc, OpBuilder &builder)
 
SmallVector< Valuemlir::tensor::createDynamicDimValues (OpBuilder &b, Location loc, Value rankedTensor)
 
FailureOr< RankedTensorType > mlir::tensor::computeTransposedType (RankedTensorType rankedTensorType, ArrayRef< int64_t > transposeVector)
 Returns the transposed rankedTensorType if transposeVector is non-empty. More...
 
SmallVector< int64_t > mlir::tensor::getPackInverseDestPerm (tensor::PackOp packOp)
 Shell function to compute the Destination Permutation of PackOp This function uses the helper function computePackUnPackPerm to get the permutation vector. More...
 
SmallVector< int64_t > mlir::tensor::getUnPackInverseSrcPerm (tensor::UnPackOp unpackOp)
 Shell function to compute the Source Permutation of unPackOp. More...
 
SmallVector< int64_t > mlir::tensor::getUnPackInverseSrcPerm (tensor::UnPackOp, PackingMetadata &metadata)
 Shell function to compute the Source rank permutation for unpackOp Unpack requires some packing metadata data information, so created another function where this value is passed by reference. More...
 
bool mlir::tensor::isCastLikeInsertSliceOp (InsertSliceOp op)
 A tensor.insert_slice is a cast-like operation if it merely rank-extends the source tensor or inserts the source tensor into a destination tensor with the same shape. More...
 
bool mlir::tensor::isCastLikeExtractSliceOp (ExtractSliceOp op)
 A tensor.extract_slice is a cast-like operation if it merely rank-reduces unit dimensions of the source tensor or extracts the entire source tensor. More...