Searched refs:LmhloDialect (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.h | 40 class LmhloDialect : public Dialect {
42 explicit LmhloDialect(MLIRContext *context);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | CMakeLists.txt | 63 add_mlir_dialect_library(LmhloDialect
69 target_link_libraries(LmhloDialect PUBLIC MLIRIR)
94 LmhloDialect
|
| D | lhlo_ops.cc | 52 LmhloDialect::LmhloDialect(MLIRContext* context) in LmhloDialect() function in mlir::lmhlo::LmhloDialect
53 : Dialect(getDialectNamespace(), context, TypeID::get<LmhloDialect>()) { in LmhloDialect()
|
| D | init.cc | 26 mlir::lmhlo::LmhloDialect, in registerAllMhloDialects()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | bufferize_pass.cc | 108 registry.insert<lmhlo::LmhloDialect>(); in getDependentDialects()
116 target.addLegalDialect<complex::ComplexDialect, lmhlo::LmhloDialect, in runOnOperation()
159 lmhlo::LmhloDialect>(); in getDependentDialects()
169 shape::ShapeDialect, lmhlo::LmhloDialect, in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | CMakeLists.txt | 94 LmhloDialect
149 LmhloDialect
|
| D | lhlo_legalize_to_gpu.cc | 183 gpu::GPUDialect, scf::SCFDialect, LmhloDialect>(); in runOnFunction()
|
| D | hlo_legalize_to_lhlo.cc | 567 registry.insert<lmhlo::LmhloDialect>(); in getDependentDialects()
578 target.addLegalDialect<lmhlo::LmhloDialect>(); in runOnOperation()
|
| D | lhlo_legalize_to_parallel_loops.cc | 716 scf::SCFDialect, LmhloDialect>(); in runOnFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tools/mlir-hlo-opt/ |
| D | mlir-hlo-opt.cpp | 34 registry.insert<mlir::lmhlo::LmhloDialect>(); in main()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 146 context.loadDialect<mlir::lmhlo::LmhloDialect, mlir::mhlo::MhloDialect, in ParseMlirModule()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 160 mlir::lmhlo::LmhloDialect, mlir::lmhlo_gpu::LmhloGpuDialect>(); in getDependentDialects()
1535 ->loadDialect<StandardOpsDialect, mhlo::MhloDialect, lmhlo::LmhloDialect, in HloToLhloModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 592 mlir_context.loadDialect<mlir::lmhlo::LmhloDialect, mlir::mhlo::MhloDialect, in CompileModuleToLlvmIrImpl()
|