Searched refs:JitRuntimeDialect (Results 1 – 5 of 5) sorted by relevance
66 JitRuntimeDialect::JitRuntimeDialect(mlir::MLIRContext* context) in JitRuntimeDialect() function in mlir::tf_jitrt::JitRuntimeDialect68 mlir::TypeID::get<JitRuntimeDialect>()) { in JitRuntimeDialect()
30 class JitRuntimeDialect : public mlir::Dialect {32 explicit JitRuntimeDialect(mlir::MLIRContext *context);
23 registry.insert<mlir::tf_jitrt::JitRuntimeDialect>(); in RegisterTfJitRtDialect()
52 registry.insert<mlir::tf_jitrt::JitRuntimeDialect>(); in main()
92 tf_jitrt::JitRuntimeDialect>(); in getDependentConversionDialects()1498 target->addLegalDialect<tf_jitrt::JitRuntimeDialect>(); in SetUpTFToTFRTConversionLegality()1501 target->addIllegalDialect<tfrt::jitrt::JitRuntimeDialect>(); in SetUpTFToTFRTConversionLegality()1980 registry.insert<tfrt::jitrt::JitRuntimeDialect>(); in getDependentDialects()