Home
last modified time | relevance | path

Searched refs:LmhloGpuDialect (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dlhlo_gpu_ops.cc50 LmhloGpuDialect::LmhloGpuDialect(MLIRContext *context) in LmhloGpuDialect() function in mlir::lmhlo_gpu::LmhloGpuDialect
51 : Dialect(getDialectNamespace(), context, TypeID::get<LmhloGpuDialect>()) { in LmhloGpuDialect()
Dinit.cc27 mlir::lmhlo_gpu::LmhloGpuDialect>(); in registerAllMhloDialects()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_gpu_ops.h48 class LmhloGpuDialect : public Dialect {
50 explicit LmhloGpuDialect(MLIRContext *context);
/external/tensorflow/tensorflow/compiler/mlir/hlo/tools/mlir-hlo-opt/
Dmlir-hlo-opt.cpp35 registry.insert<mlir::lmhlo_gpu::LmhloGpuDialect>(); in main()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc148 mlir::lmhlo_gpu::LmhloGpuDialect>(); in ParseMlirModule()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc160 mlir::lmhlo::LmhloDialect, mlir::lmhlo_gpu::LmhloGpuDialect>(); in getDependentDialects()
1536 lmhlo_gpu::LmhloGpuDialect>(); in HloToLhloModule()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc594 mlir::lmhlo_gpu::LmhloGpuDialect>(); in CompileModuleToLlvmIrImpl()