| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 331 sparse_tensor::ConcatenateOp::attachInterface<ConcatenateOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
332 sparse_tensor::ConvertOp::attachInterface<ConvertOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
333 sparse_tensor::LoadOp::attachInterface<LoadOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
334 sparse_tensor::NewOp::attachInterface<NewOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
335 sparse_tensor::NumberOfEntriesOp::attachInterface< in registerBufferizableOpInterfaceExternalModels()
337 sparse_tensor::AssembleOp::attachInterface<AssembleOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
338 sparse_tensor::DisassembleOp::attachInterface<DisassembleOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
339 sparse_tensor::ForeachOp::attachInterface<ForeachOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
340 sparse_tensor::ToCoordinatesBufferOp::attachInterface< in registerBufferizableOpInterfaceExternalModels()
342 sparse_tensor::ToCoordinatesOp::attachInterface<ToCoordinatesOpInterface>( in registerBufferizableOpInterfaceExternalModels()
[all …]
|
| /llvm-project/mlir/unittests/IR/ |
| H A D | InterfaceAttachmentTest.cpp | 68 IntegerType::attachInterface<Model>(context); in TEST()
79 Float32Type::attachInterface<OverridingModel>(context); in TEST()
107 test::TestType::attachInterface<TestTypeModel>(*ctx); in TEST()
126 test::TestType::attachInterface<TestTypeModel>(*ctx); in TEST()
142 IntegerType::attachInterface<Model>(*ctx); in TEST()
155 IntegerType::attachInterface<Model>(*ctx); in TEST()
197 IntegerType::attachInterface<TestExternalFallbackTypeIntegerModel>(context); in TEST()
203 VectorType::attachInterface<TestExternalFallbackTypeVectorModel>(context); in TEST()
226 IntegerAttr::attachInterface<TestExternalIntegerAttrModel>(context); in TEST()
250 test::SimpleAAttr::attachInterface<TestExternalSimpleAAttrMode in TEST()
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | SubsetInsertionOpInterfaceImpl.cpp | 92 ExtractSliceOp::attachInterface<ExtractSliceOpSubsetOpInterface>(*ctx); in registerSubsetOpInterfaceExternalModels()
93 ExtractSliceOp::attachInterface<ExtractSliceOpSubsetExtractionOpInterface>( in registerSubsetOpInterfaceExternalModels()
95 InsertSliceOp::attachInterface< in registerSubsetOpInterfaceExternalModels()
97 InsertSliceOp::attachInterface< in registerSubsetOpInterfaceExternalModels()
99 ParallelInsertSliceOp::attachInterface< in registerSubsetOpInterfaceExternalModels()
101 ParallelInsertSliceOp::attachInterface< in registerSubsetOpInterfaceExternalModels()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 1057 CastOp::attachInterface<CastOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1058 CollapseShapeOp::attachInterface<CollapseShapeOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1059 DimOp::attachInterface<DimOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1060 EmptyOp::attachInterface<EmptyOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1061 ExpandShapeOp::attachInterface<ExpandShapeOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1062 ExtractSliceOp::attachInterface<ExtractSliceOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1063 ExtractOp::attachInterface<ExtractOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1064 FromElementsOp::attachInterface<FromElementsOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1065 GenerateOp::attachInterface<GenerateOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
1066 InsertOp::attachInterface<InsertOpInterfac in registerBufferizableOpInterfaceExternalModels()
[all...] |
| /llvm-project/mlir/lib/Dialect/MemRef/IR/ |
| H A D | ValueBoundsOpInterfaceImpl.cpp | 120 memref::AllocOp::attachInterface<memref::AllocOpInterface<memref::AllocOp>>( in registerValueBoundsOpInterfaceExternalModels()
122 memref::AllocaOp::attachInterface< in registerValueBoundsOpInterfaceExternalModels()
124 memref::CastOp::attachInterface<memref::CastOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
125 memref::DimOp::attachInterface<memref::DimOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
126 memref::GetGlobalOp::attachInterface<memref::GetGlobalOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
127 memref::RankOp::attachInterface<memref::RankOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
128 memref::SubViewOp::attachInterface<memref::SubViewOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | ValueBoundsOpInterfaceImpl.cpp | 119 tensor::CastOp::attachInterface<tensor::CastOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
120 tensor::DimOp::attachInterface<tensor::DimOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
121 tensor::EmptyOp::attachInterface<tensor::EmptyOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
122 tensor::ExtractSliceOp::attachInterface<tensor::ExtractSliceOpInterface>( in registerValueBoundsOpInterfaceExternalModels()
124 tensor::PadOp::attachInterface<tensor::PadOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
125 tensor::RankOp::attachInterface<tensor::RankOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
|
| H A D | TensorInferTypeOpInterfaceImpl.cpp | 132 ExpandShapeOp::attachInterface<ReifyExpandShapeOp>(*ctx); in getExpandedOutputShapeFromInputShape()
133 CollapseShapeOp::attachInterface<ReifyCollapseShapeOp>(*ctx);
134 PadOp::attachInterface<ReifyPadOp>(*ctx);
|
| /llvm-project/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | SubsetOpInterfaceImpl.cpp | 73 TransferReadOp::attachInterface<XferOpSubsetOpInterface<TransferReadOp>>( in registerSubsetOpInterfaceExternalModels()
75 TransferReadOp::attachInterface<TransferReadOpSubsetExtractionOpInterface>( in registerSubsetOpInterfaceExternalModels()
77 TransferWriteOp::attachInterface<XferOpSubsetOpInterface<TransferWriteOp>>( in registerSubsetOpInterfaceExternalModels()
79 TransferWriteOp::attachInterface<TransferWriteOpSubsetInsertionOpInterface>( in registerSubsetOpInterfaceExternalModels()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 321 TransferReadOp::attachInterface<TransferReadOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
322 TransferWriteOp::attachInterface<TransferWriteOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
323 GatherOp::attachInterface<GatherOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
324 MaskOp::attachInterface<MaskOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
325 YieldOp::attachInterface<YieldOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/Arith/IR/ |
| H A D | ValueBoundsOpInterfaceImpl.cpp | 156 arith::AddIOp::attachInterface<arith::AddIOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
157 arith::ConstantOp::attachInterface<arith::ConstantOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
158 arith::SubIOp::attachInterface<arith::SubIOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
159 arith::MulIOp::attachInterface<arith::MulIOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
160 arith::SelectOp::attachInterface<arith::SelectOpInterface>(*ctx); in registerValueBoundsOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/Tensor/TransformOps/ |
| H A D | TensorTransformOps.cpp | 74 CollapseShapeOp::attachInterface< in registerFindPayloadReplacementOpInterfaceExternalModels()
76 ExpandShapeOp::attachInterface< in registerFindPayloadReplacementOpInterfaceExternalModels()
78 ExtractSliceOp::attachInterface<ExtractSliceOpReplacementInterface>(*ctx); in registerFindPayloadReplacementOpInterfaceExternalModels()
79 InsertSliceOp::attachInterface<InsertSliceOpReplacementInterface>(*ctx); in registerFindPayloadReplacementOpInterfaceExternalModels()
80 ReshapeOp::attachInterface<ReshapeOpReplacementInterface>(*ctx); in registerFindPayloadReplacementOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | AllocationOpInterfaceImpl.cpp | 64 memref::AllocOp::attachInterface<DefaultAllocationInterface>(*ctx); in registerAllocationOpInterfaceExternalModels()
65 memref::AllocaOp::attachInterface< in registerAllocationOpInterfaceExternalModels()
67 memref::ReallocOp::attachInterface<DefaultReallocationInterface>(*ctx); in registerAllocationOpInterfaceExternalModels()
|
| H A D | RuntimeOpVerification.cpp | 337 CastOp::attachInterface<CastOpInterface>(*ctx); in registerRuntimeVerifiableOpInterfaceExternalModels()
338 ExpandShapeOp::attachInterface<ExpandShapeOpInterface>(*ctx); in registerRuntimeVerifiableOpInterfaceExternalModels()
339 LoadOp::attachInterface<LoadStoreOpInterface<LoadOp>>(*ctx); in registerRuntimeVerifiableOpInterfaceExternalModels()
340 ReinterpretCastOp::attachInterface<ReinterpretCastOpInterface>(*ctx); in registerRuntimeVerifiableOpInterfaceExternalModels()
341 StoreOp::attachInterface<LoadStoreOpInterface<StoreOp>>(*ctx); in registerRuntimeVerifiableOpInterfaceExternalModels()
342 SubViewOp::attachInterface<SubViewOpInterface>(*ctx); in registerRuntimeVerifiableOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/MLProgram/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 153 GlobalOp::attachInterface<GlobalOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
154 GlobalLoadOp::attachInterface<GlobalLoadOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
155 GlobalStoreOp::attachInterface<GlobalStoreOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | RuntimeOpVerification.cpp | 115 void attachInterface(MLIRContext *ctx) { in attachInterface() function
116 (OpTs::template attachInterface<StructuredOpInterface<OpTs>>(*ctx), ...); in attachInterface()
125 attachInterface< in registerRuntimeVerifiableOpInterfaceExternalModels()
|
| H A D | SubsetInsertionOpInterfaceImpl.cpp | 78 linalg::CopyOp::attachInterface<LinalgCopyOpSubsetOpInterface>(*ctx); in registerSubsetOpInterfaceExternalModels()
79 linalg::CopyOp::attachInterface<LinalgCopyOpInterface>(*ctx); in registerSubsetOpInterfaceExternalModels()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 162 (Ops::template attachInterface<LinalgOpInterface<Ops>>(*ctx), ...); in registerOpInterface()
207 SoftmaxOp::attachInterface<SoftmaxOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/ControlFlow/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 67 cf::BranchOp::attachInterface<BranchOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
68 cf::CondBranchOp::attachInterface<CondBranchOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | BufferDeallocationOpInterfaceImpl.cpp | 79 InParallelOp::attachInterface<InParallelOpInterface>(*ctx); in registerBufferDeallocationOpInterfaceExternalModels()
80 ReduceReturnOp::attachInterface<ReduceReturnOpInterface>(*ctx); in registerBufferDeallocationOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/Affine/IR/ |
| H A D | ValueBoundsOpInterfaceImpl.cpp | 158 AffineApplyOp::attachInterface<AffineApplyOpInterface>(*ctx);
159 AffineMaxOp::attachInterface<AffineMaxOpInterface>(*ctx);
160 AffineMinOp::attachInterface<AffineMinOpInterface>(*ctx);
161 AffineDelinearizeIndexOp::attachInterface<
163 AffineLinearizeIndexOp::attachInterface<AffineLinearizeIndexOpInterface>(
|
| /llvm-project/mlir/lib/Dialect/Tosa/IR/ |
| H A D | ShardingInterfaceImpl.cpp | 65 OpType::template attachInterface<ElementwiseShardingInterface<OpType>>(*ctx); in registerElemwiseOne()
88 MatMulOp::attachInterface<MatMulOpSharding>(*ctx); in registerShardingInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/Arith/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 210 ConstantOp::attachInterface<ConstantOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
211 IndexCastOp::attachInterface<IndexCastOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
212 SelectOp::attachInterface<SelectOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/Shape/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 141 shape::AssumingOp::attachInterface<AssumingOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
142 shape::AssumingYieldOp::attachInterface<AssumingYieldOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
|
| /llvm-project/mlir/lib/Dialect/SCF/IR/ |
| H A D | ValueBoundsOpInterfaceImpl.cpp | 220 scf::ForOp::attachInterface<scf::ForOpInterface>(*ctx);
221 scf::ForallOp::attachInterface<scf::ForallOpInterface>(*ctx);
222 scf::IfOp::attachInterface<scf::IfOpInterface>(*ctx);
|
| /llvm-project/mlir/lib/Dialect/Func/Extensions/ |
| H A D | MeshShardingExtensions.cpp | 18 ReturnOp::attachInterface< in registerShardingInterfaceExternalModels()
|