Searched refs:custom_legalization_passes (Results 1 – 3 of 3) sorted by relevance
42 custom_legalization_passes);76 custom_legalization_passes = {});97 custom_legalization_passes);118 custom_legalization_passes);128 custom_legalization_passes = {});141 custom_legalization_passes);153 custom_legalization_passes = {});168 custom_legalization_passes = {});
281 custom_legalization_passes) { in CreateConvertMlirToXlaHloPipeline()314 for (auto& target_pass : custom_legalization_passes) { in CreateConvertMlirToXlaHloPipeline()343 custom_legalization_passes) { in LegalizeToHlo()347 custom_legalization_passes); in LegalizeToHlo()380 custom_legalization_passes) { in BuildHloFromTfInner() argument382 LegalizeToHlo(module_op, device_type, custom_legalization_passes)); in BuildHloFromTfInner()394 custom_legalization_passes) { in ConvertMLIRToXlaComputation()396 LegalizeToHlo(module_op, device_type, custom_legalization_passes)); in ConvertMLIRToXlaComputation()427 custom_legalization_passes) { in BuildHloFromTf() argument438 custom_legalization_passes)); in BuildHloFromTf()[all …]
244 custom_legalization_passes) { in CompileMlirToXlaHloViaBuilder()267 custom_legalization_passes)); in CompileMlirToXlaHloViaBuilder()307 custom_legalization_passes{}; in MlirTfToHloTextTranslateFunctionImpl() local313 custom_legalization_passes) in MlirTfToHloTextTranslateFunctionImpl()319 custom_legalization_passes); in MlirTfToHloTextTranslateFunctionImpl()