| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | CMakeLists.txt | 44 add_mlir_dialect_library(MhloDialect
55 target_link_libraries(MhloDialect
93 MhloDialect
|
| D | init.cc | 25 mlir::mhlo::MhloDialect, in registerAllMhloDialects()
|
| D | hlo_ops.cc | 70 Operation* MhloDialect::materializeConstant(OpBuilder& builder, Attribute value, in materializeConstant()
3134 MhloDialect::MhloDialect(MLIRContext* context) in MhloDialect() function in mlir::mhlo::MhloDialect
3135 : Dialect(getDialectNamespace(), context, TypeID::get<MhloDialect>()) { in MhloDialect()
3145 Type MhloDialect::parseType(DialectAsmParser& parser) const { in parseType()
3154 void MhloDialect::printType(Type type, DialectAsmPrinter& os) const { in printType()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.h | 46 class MhloDialect : public Dialect {
48 explicit MhloDialect(MLIRContext *context);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo_pass.cc | 41 registry.insert<mhlo::MhloDialect, shape::ShapeDialect, scf::SCFDialect>(); in getDependentDialects()
52 MhloDialect, mlir::StandardOpsDialect, mlir::tensor::TensorDialect, in runOnFunction()
|
| D | CMakeLists.txt | 93 MhloDialect
130 MhloDialect
|
| D | materialize_broadcasts_pass.cc | 37 conversionTarget.addLegalDialect<MhloDialect>(); in runOnFunction()
|
| D | transform_unranked_hlo.cc | 469 registry.insert<shape::ShapeDialect, mhlo::MhloDialect>(); in getDependentDialects()
476 target.addLegalDialect<mhlo::MhloDialect, StandardOpsDialect, in runOnFunction()
|
| D | hlo_legalize_to_lhlo.cc | 581 target.addIllegalDialect<mhlo::MhloDialect>(); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tools/mlir-hlo-opt/ |
| D | mlir-hlo-opt.cpp | 32 registry.insert<mlir::mhlo::MhloDialect>(); in main()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | bufferize_pass.cc | 119 target.addIllegalDialect<mhlo::MhloDialect>(); in runOnOperation()
173 target.addIllegalDialect<mhlo::MhloDialect>(); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_module_importer.cc | 39 module.getContext()->loadDialect<mlir::mhlo::MhloDialect>(); in HloModuleImporter()
|
| D | hlo_function_importer.h | 89 context_->loadDialect<mlir::mhlo::MhloDialect>(); in HloFunctionImporter()
|
| D | xla_mlir_translate.cc | 260 registry.insert<mlir::StandardOpsDialect, mlir::mhlo::MhloDialect, in RegisterInputDialects()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 48 context_.loadDialect<mlir::mhlo::MhloDialect>(); in XlaBuilderTest()
|
| /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/tools/kernel_gen/ |
| D | kernel_creator.cc | 437 registry.insert<mlir::chlo::HloClientDialect, mlir::mhlo::MhloDialect>(); in GenerateKernelForTfCode()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 64 registry.insert<mhlo::MhloDialect>(); in getDependentDialects()
|
| D | mhlo_to_lhlo_with_xla.cc | 159 .insert<mlir::StandardOpsDialect, mlir::mhlo::MhloDialect, in getDependentDialects()
1535 ->loadDialect<StandardOpsDialect, mhlo::MhloDialect, lmhlo::LmhloDialect, in HloToLhloModule()
|
| D | legalize_tf.cc | 76 registry.insert<chlo::HloClientDialect, mhlo::MhloDialect, in getDependentDialects()
6186 target.addLegalDialect<MhloDialect>(); in legalizeTF()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 36 return mlir::mhlo::MhloDialect::getDialectNamespace().str() + "." + op_name; in GetMlirOpName()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 592 mlir_context.loadDialect<mlir::lmhlo::LmhloDialect, mlir::mhlo::MhloDialect, in CompileModuleToLlvmIrImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 95 registry.insert<mhlo::MhloDialect, quant::QuantizationDialect, in getDependentDialects()
|