Home
last modified time | relevance | path

Searched refs:addDynamicallyLegalOp (Results 1 – 23 of 23) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/
DFuncBufferize.cpp35 target.addDynamicallyLegalOp<FuncOp>([&](FuncOp op) { in runOnOperation()
40 target.addDynamicallyLegalOp<CallOp>( in runOnOperation()
47 target.addDynamicallyLegalOp<ReturnOp>( in runOnOperation()
DBufferize.cpp193 target.addDynamicallyLegalOp<SelectOp>([&](SelectOp op) { in runOnFunction()
197 target.addDynamicallyLegalOp<DimOp>( in runOnFunction()
DTensorConstantBufferize.cpp114 target.addDynamicallyLegalOp<ConstantOp>( in runOnOperation()
DExpandOps.cpp222 target.addDynamicallyLegalOp<AtomicRMWOp>([](AtomicRMWOp op) { in runOnFunction()
226 target.addDynamicallyLegalOp<MemRefReshapeOp>([](MemRefReshapeOp op) { in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dembed_tf_framework_pass.cc60 target.addDynamicallyLegalOp<FuncOp>([&](FuncOp op) { in runOnOperation()
68 target.addDynamicallyLegalOp<AllocOp, DeallocOp>( in runOnOperation()
101 target.addDynamicallyLegalOp<AssertOp>(IsNotInsideTfEntryFunction); in runOnOperation()
Dbufferize_pass.cc138 target.addDynamicallyLegalOp<FuncOp>([&](FuncOp op) { in runOnOperation()
146 target.addDynamicallyLegalOp<CallOp, ReturnOp>(isLegalOp); in runOnOperation()
182 target.addDynamicallyLegalOp<ConstantOp, DimOp, RankOp, SelectOp>( in runOnOperation()
/external/llvm-project/mlir/test/lib/Transforms/
DTestDecomposeCallGraphTypes.cpp40 target.addDynamicallyLegalOp<ReturnOp>([&](ReturnOp op) { in runOnOperation()
43 target.addDynamicallyLegalOp<CallOp>( in runOnOperation()
45 target.addDynamicallyLegalOp<FuncOp>([&](FuncOp op) { in runOnOperation()
/external/llvm-project/mlir/include/mlir/Transforms/
DDialectConversion.h589 template <typename OpT> void addDynamicallyLegalOp() { in addDynamicallyLegalOp() function
593 void addDynamicallyLegalOp() { in addDynamicallyLegalOp() function
594 addDynamicallyLegalOp<OpT>(); in addDynamicallyLegalOp()
595 addDynamicallyLegalOp<OpT2, OpTs...>(); in addDynamicallyLegalOp()
601 void addDynamicallyLegalOp(const DynamicLegalityCallbackFn &callback) { in addDynamicallyLegalOp() function
607 void addDynamicallyLegalOp(const DynamicLegalityCallbackFn &callback) { in addDynamicallyLegalOp() function
608 addDynamicallyLegalOp<OpT>(callback); in addDynamicallyLegalOp()
609 addDynamicallyLegalOp<OpT2, OpTs...>(callback); in addDynamicallyLegalOp()
614 addDynamicallyLegalOp(Callable &&callback) { in addDynamicallyLegalOp() function
615 addDynamicallyLegalOp<OpT>( in addDynamicallyLegalOp()
/external/llvm-project/mlir/test/lib/Dialect/Test/
DTestPatterns.cpp579 target.addDynamicallyLegalOp<TestReturnOp>([](TestReturnOp op) { in runOnOperation()
584 target.addDynamicallyLegalOp<FuncOp>([&](FuncOp op) { in runOnOperation()
590 target.addDynamicallyLegalOp<TestTypeProducerOp>( in runOnOperation()
592 target.addDynamicallyLegalOp<TestTypeConsumerOp>([](TestTypeConsumerOp op) { in runOnOperation()
603 target.addDynamicallyLegalOp<TestRecursiveRewriteOp>( in runOnOperation()
707 target.addDynamicallyLegalOp<OneVResOneVOperandOp1>( in runOnFunction()
825 target.addDynamicallyLegalOp<TestTypeProducerOp>([](TestTypeProducerOp op) { in runOnOperation()
828 target.addDynamicallyLegalOp<FuncOp>([&](FuncOp op) { in runOnOperation()
832 target.addDynamicallyLegalOp<TestCastOp>([&](TestCastOp op) { in runOnOperation()
932 target.addDynamicallyLegalOp<TestMergeBlocksOp>( in runOnOperation()
[all …]
/external/llvm-project/mlir/lib/Dialect/Shape/Transforms/
DStructuralTypeConversions.cpp64 target.addDynamicallyLegalOp<AssumingOp>([&](AssumingOp op) { in populateShapeStructuralTypeConversionsAndLegality()
67 target.addDynamicallyLegalOp<AssumingYieldOp>([&](AssumingYieldOp op) { in populateShapeStructuralTypeConversionsAndLegality()
/external/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/
DDecorateSPIRVCompositeTypeLayoutPass.cpp98 target.addDynamicallyLegalOp<spirv::GlobalVariableOp>( in runOnOperation()
104 target.addDynamicallyLegalOp<spirv::AddressOfOp>([](spirv::AddressOfOp op) { in runOnOperation()
DLowerABIAttributesPass.cpp253 target.addDynamicallyLegalOp<spirv::FuncOp>([&](spirv::FuncOp op) { in runOnOperation()
/external/llvm-project/mlir/lib/Dialect/SCF/Transforms/
DStructuralTypeConversions.cpp141 target.addDynamicallyLegalOp<ForOp, IfOp>([&](Operation *op) { in populateSCFStructuralTypeConversionsAndLegality()
144 target.addDynamicallyLegalOp<scf::YieldOp>([&](scf::YieldOp op) { in populateSCFStructuralTypeConversionsAndLegality()
/external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/
DLinalgToSPIRVPass.cpp39 target->addDynamicallyLegalOp<FuncOp>([&](FuncOp op) { in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmaterialize_broadcasts.cc75 conversionTarget->addDynamicallyLegalOp<ClampOp>([](ClampOp op) { in SetupMaterializeBroadcastsLegality()
Dhlo_legalize_to_lhlo.cc586 target.addDynamicallyLegalOp<mlir::TensorToMemrefOp>( in runOnOperation()
591 target.addDynamicallyLegalOp<FuncOp>([&](FuncOp op) { in runOnOperation()
596 target.addDynamicallyLegalOp<CallOp>([&](CallOp op) { in runOnOperation()
602 target.addDynamicallyLegalOp<mlir::ReturnOp>([&](mlir::ReturnOp op) { in runOnOperation()
Dtransform_unranked_hlo.cc65 target->addDynamicallyLegalOp<OpTy>([](OpTy op) { in AddLegalOpOnRankedTensor()
/external/llvm-project/mlir/lib/Conversion/OpenMPToLLVM/
DOpenMPToLLVM.cpp72 target.addDynamicallyLegalOp<omp::ParallelOp, omp::WsLoopOp>( in runOnOperation()
/external/llvm-project/mlir/lib/Conversion/SCFToOpenMP/
DSCFToOpenMP.cpp89 target.addDynamicallyLegalOp<scf::YieldOp>( in applyPatterns()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DBufferize.cpp337 target.addDynamicallyLegalOp<ConstantOp>(isLegalOperation); in runOnOperation()
/external/llvm-project/mlir/lib/Conversion/AsyncToLLVM/
DAsyncToLLVM.cpp873 target.addDynamicallyLegalOp<FuncOp>( in runOnOperation()
875 target.addDynamicallyLegalOp<CallOp>( in runOnOperation()
/external/llvm-project/mlir/docs/
DDialectConversion.md100 addDynamicallyLegalOp<ReturnOp>();
104 addDynamicallyLegalOp<ReturnOp>([](ReturnOp op) { ... });
141 target.addDynamicallyLegalOp<MySecondOp>(...);
/external/llvm-project/mlir/lib/Conversion/SCFToGPU/
DSCFToGPU.cpp650 target.addDynamicallyLegalOp<scf::ParallelOp>([](scf::ParallelOp parallelOp) { in configureParallelLoopToGPULegality()