Changeset View
Changeset View
Standalone View
Standalone View
mlir/lib/Dialect/Shape/Transforms/ShapeTypeConversion.cpp
Show All 12 Lines | |||||
#include "PassDetail.h" | #include "PassDetail.h" | ||||
#include "mlir/Dialect/Shape/IR/Shape.h" | #include "mlir/Dialect/Shape/IR/Shape.h" | ||||
#include "mlir/Dialect/Shape/Transforms/Passes.h" | #include "mlir/Dialect/Shape/Transforms/Passes.h" | ||||
#include "mlir/Dialect/StandardOps/IR/Ops.h" | #include "mlir/Dialect/StandardOps/IR/Ops.h" | ||||
#include "mlir/IR/Operation.h" | #include "mlir/IR/Operation.h" | ||||
#include "mlir/IR/StandardTypes.h" | #include "mlir/IR/StandardTypes.h" | ||||
#include "mlir/Pass/Pass.h" | #include "mlir/Pass/Pass.h" | ||||
#include "mlir/Transforms/BufferPlacement.h" | #include "mlir/Transforms/Bufferize.h" | ||||
using namespace mlir; | using namespace mlir; | ||||
using namespace mlir::shape; | using namespace mlir::shape; | ||||
namespace { | namespace { | ||||
// Propagate tensor to memref conversions through shape.assuming ops. | // Propagate tensor to memref conversions through shape.assuming ops. | ||||
class TypeConversionAssumingOpConverter | class TypeConversionAssumingOpConverter | ||||
: public BufferAssignmentOpConversionPattern<shape::AssumingOp> { | : public BufferAssignmentOpConversionPattern<shape::AssumingOp> { | ||||
▲ Show 20 Lines • Show All 66 Lines • Show Last 20 Lines |