MLIR  20.0.0git
Namespaces | Macros | Typedefs | Functions
Tensor.h File Reference
#include "mlir/Bytecode/BytecodeOpInterface.h"
#include "mlir/Dialect/Utils/ReshapeOpsUtils.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Dialect.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/IR/OpImplementation.h"
#include "mlir/Interfaces/CastInterfaces.h"
#include "mlir/Interfaces/ControlFlowInterfaces.h"
#include "mlir/Interfaces/DestinationStyleOpInterface.h"
#include "mlir/Interfaces/InferTypeOpInterface.h"
#include "mlir/Interfaces/ParallelCombiningOpInterface.h"
#include "mlir/Interfaces/ShapedOpInterfaces.h"
#include "mlir/Interfaces/SideEffectInterfaces.h"
#include "mlir/Interfaces/TilingInterface.h"
#include "mlir/Interfaces/ViewLikeInterface.h"
#include "mlir/Dialect/Tensor/IR/TensorOpsDialect.h.inc"
#include "mlir/Dialect/Tensor/IR/TensorOps.h.inc"

Go to the source code of this file.

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::tensor
 

Macros

#define GET_OP_CLASSES
 

Typedefs

using mlir::tensor::ControlConstantExtractSliceFusionFn = std::function< bool(ExtractSliceOp)>
 Function to control the folding of constant and extract slice. More...
 

Functions

SmallVector< Range, 8 > mlir::getOrCreateRanges (OffsetSizeAndStrideOpInterface op, OpBuilder &b, Location loc)
 Return the list of Range (i.e. More...
 
bool mlir::tensor::preservesStaticInformation (Type source, Type target)
 Returns true if target is a ranked tensor type that preserves static information available in the source ranked tensor type. More...
 
bool mlir::tensor::canFoldIntoConsumerOp (CastOp castOp)
 Determines whether tensor::CastOp casts to a more dynamic version of the source tensor. More...
 
bool mlir::tensor::canFoldIntoProducerOp (CastOp castOp)
 Determines whether the tensor::CastOp casts to a more static version of the source tensor. More...
 
LogicalResult mlir::tensor::foldTensorCast (Operation *op)
 Performs folding of any operand of op if it comes from a tensor::CastOp that can be folded. More...
 
OpFoldResult mlir::tensor::getMixedSize (OpBuilder &builder, Location loc, Value value, int64_t dim)
 Return the dimension of the given tensor value. More...
 
SmallVector< OpFoldResultmlir::tensor::getMixedSizes (OpBuilder &builder, Location loc, Value value)
 Return the dimensions of the given tensor value. More...
 
Value mlir::tensor::createCanonicalRankReducingExtractSliceOp (OpBuilder &b, Location loc, Value tensor, RankedTensorType targetType)
 Create a rank-reducing ExtractSliceOp @[0 . More...
 
Value mlir::tensor::createCanonicalRankReducingInsertSliceOp (OpBuilder &b, Location loc, Value tensor, Value dest)
 Create a rank-reducing InsertSliceOp @[0 . More...
 
FailureOr< Valuemlir::tensor::getOrCreateDestination (OpBuilder &b, Location loc, OpResult opResult)
 This is a helper function for DestinationStyleOpInterface. More...
 
LogicalResult mlir::tensor::getOrCreateDestinations (OpBuilder &b, Location loc, Operation *op, SmallVector< Value > &result)
 This is a helper function for DestinationStyleOpInterface. More...
 
bool mlir::tensor::isSameTypeWithoutEncoding (Type tp1, Type tp2)
 Tests if types are the same when ignoring encoding on ranked tensors. More...
 
void mlir::tensor::populateFoldConstantExtractSlicePatterns (RewritePatternSet &patterns, const ControlConstantExtractSliceFusionFn &controlFn=[](ExtractSliceOp op) { return false;})
 Patterns to fold the extract slice op with its constant operand. More...
 

Macro Definition Documentation

◆ GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 52 of file Tensor.h.