| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | shape_to_descriptors_pass.cc | 52 target.addLegalDialect<scf::SCFDialect>(); in runOnOperation()
53 target.addLegalDialect<StandardOpsDialect>(); in runOnOperation()
54 target.addLegalDialect<math::MathDialect>(); in runOnOperation()
55 target.addLegalDialect<tensor::TensorDialect>(); in runOnOperation()
|
| D | embed_tf_framework_pass.cc | 58 target.addLegalDialect<tf_framework::TFFrameworkDialect>(); in runOnOperation()
98 target.addLegalDialect<tf_framework::TFFrameworkDialect, in runOnOperation()
|
| D | bufferize_pass.cc | 116 target.addLegalDialect<complex::ComplexDialect, lmhlo::LmhloDialect, in runOnOperation()
166 target.addLegalDialect<complex::ComplexDialect, scf::SCFDialect, in runOnOperation()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPUPass.cpp | 52 target.addLegalDialect<StandardOpsDialect>(); in runOnOperation()
53 target.addLegalDialect<AffineDialect>(); in runOnOperation()
54 target.addLegalDialect<gpu::GPUDialect>(); in runOnOperation()
55 target.addLegalDialect<scf::SCFDialect>(); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | materialize_broadcasts_pass.cc | 37 conversionTarget.addLegalDialect<MhloDialect>(); in runOnFunction()
39 conversionTarget.addLegalDialect<mlir::StandardOpsDialect>(); in runOnFunction()
|
| D | chlo_legalize_to_hlo_pass.cc | 51 conversionTarget.addLegalDialect< in runOnFunction()
|
| D | hlo_legalize_to_lhlo.cc | 578 target.addLegalDialect<lmhlo::LmhloDialect>(); in runOnOperation()
579 target.addLegalDialect<StandardOpsDialect>(); in runOnOperation()
580 target.addLegalDialect<tensor::TensorDialect>(); in runOnOperation()
|
| D | lhlo_legalize_to_gpu.cc | 182 target.addLegalDialect<linalg::LinalgDialect, StandardOpsDialect, in runOnFunction()
|
| /external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/ |
| D | ConvertSPIRVToLLVMPass.cpp | 49 target.addLegalDialect<LLVM::LLVMDialect>(); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToNVVM/ |
| D | LowerGpuOpsToNVVMOps.cpp | 148 target.addLegalDialect<::mlir::LLVM::LLVMDialect>(); in configureGpuToNVVMConversionLegality()
149 target.addLegalDialect<::mlir::NVVM::NVVMDialect>(); in configureGpuToNVVMConversionLegality()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestConvertCallOp.cpp | 58 target.addLegalDialect<LLVM::LLVMDialect>(); in runOnOperation()
|
| D | TestDecomposeCallGraphTypes.cpp | 38 target.addLegalDialect<test::TestDialect>(); in runOnOperation()
|
| D | TestConvVectorization.cpp | 56 target.addLegalDialect<AffineDialect, scf::SCFDialect, StandardOpsDialect, in runOnOperation()
|
| /external/llvm-project/mlir/lib/Conversion/VectorToLLVM/ |
| D | ConvertVectorToLLVMPass.cpp | 60 target.addLegalDialect<LLVM::LLVMAVX512Dialect>(); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/ |
| D | Bufferize.cpp | 184 target.addLegalDialect<StandardOpsDialect>(); in runOnFunction()
185 target.addLegalDialect<scf::SCFDialect>(); in runOnFunction()
|
| D | TensorConstantBufferize.cpp | 112 target.addLegalDialect<StandardOpsDialect>(); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Dialect/Shape/Transforms/ |
| D | ShapeToShapeLowering.cpp | 68 target.addLegalDialect<ShapeDialect, StandardOpsDialect>(); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToOpenMP/ |
| D | SCFToOpenMP.cpp | 91 target.addLegalDialect<omp::OpenMPDialect>(); in applyPatterns()
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/ |
| D | DecorateSPIRVCompositeTypeLayoutPass.cpp | 96 target.addLegalDialect<spirv::SPIRVDialect>(); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToROCDL/ |
| D | LowerGpuOpsToROCDLOps.cpp | 77 target.addLegalDialect<ROCDL::ROCDLDialect>(); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Conversion/VectorToROCDL/ |
| D | VectorToROCDL.cpp | 174 target.addLegalDialect<ROCDL::ROCDLDialect>(); in runOnOperation()
|
| /external/llvm-project/mlir/examples/toy/Ch6/mlir/ |
| D | LowerToAffineLoops.cpp | 286 target.addLegalDialect<AffineDialect, StandardOpsDialect>(); in runOnFunction()
|
| /external/llvm-project/mlir/examples/toy/Ch7/mlir/ |
| D | LowerToAffineLoops.cpp | 287 target.addLegalDialect<AffineDialect, StandardOpsDialect>(); in runOnFunction()
|
| /external/llvm-project/mlir/examples/toy/Ch5/mlir/ |
| D | LowerToAffineLoops.cpp | 287 target.addLegalDialect<AffineDialect, StandardOpsDialect>(); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.cpp | 210 target.addLegalDialect<AffineDialect, scf::SCFDialect, StandardOpsDialect>(); in runOnOperation()
|