Searched refs:IrExportStage (Results 1 – 3 of 3) sorted by relevance
50 IrExportStage stage, ProcessFunctionLibraryRuntime* pflr, in GetCompilerIr()122 case IrExportStage::HLO: in GetCompilerIr()123 case IrExportStage::HLO_SERIALIZED: { in GetCompilerIr()135 if (stage == IrExportStage::HLO_SERIALIZED) { in GetCompilerIr()141 case IrExportStage::OPTIMIZED_HLO: in GetCompilerIr()142 case IrExportStage::OPTIMIZED_HLO_SERIALIZED: { in GetCompilerIr()147 if (stage == IrExportStage::OPTIMIZED_HLO_SERIALIZED) { in GetCompilerIr()153 case IrExportStage::OPTIMIZED_HLO_DOT: { in GetCompilerIr()
30 enum class IrExportStage { enum41 IrExportStage stage, ProcessFunctionLibraryRuntime* pflr,
309 IrExportStage selected_stage = [&] { in InputTFE_OutputTensorHandles()311 return IrExportStage::HLO; in InputTFE_OutputTensorHandles()313 return IrExportStage::HLO_SERIALIZED; in InputTFE_OutputTensorHandles()315 return IrExportStage::OPTIMIZED_HLO; in InputTFE_OutputTensorHandles()317 return IrExportStage::OPTIMIZED_HLO_SERIALIZED; in InputTFE_OutputTensorHandles()319 return IrExportStage::OPTIMIZED_HLO_DOT; in InputTFE_OutputTensorHandles()