Searched refs:HloClientDialect (Results 1 – 8 of 8) sorted by relevance
35 class HloClientDialect : public Dialect {39 explicit HloClientDialect(MLIRContext* context) in HloClientDialect() function41 TypeID::get<HloClientDialect>()) { in HloClientDialect()
24 registry.insert<mlir::chlo::HloClientDialect, in registerAllMhloDialects()
389 void HloClientDialect::initialize() { in initialize()
33 registry.insert<mlir::chlo::HloClientDialect>(); in main()
47 conversionTarget.addIllegalDialect<chlo::HloClientDialect>(); in runOnFunction()
489 target.addDynamicallyLegalDialect<chlo::HloClientDialect>( in runOnFunction()
437 registry.insert<mlir::chlo::HloClientDialect, mlir::mhlo::MhloDialect>(); in GenerateKernelForTfCode()
76 registry.insert<chlo::HloClientDialect, mhlo::MhloDialect, in getDependentDialects()6182 target.addIllegalDialect<chlo::HloClientDialect>(); in legalizeTF()6184 target.addLegalDialect<chlo::HloClientDialect>(); in legalizeTF()