Home
last modified time | relevance | path

Searched refs:IrExportStage (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dget_compiler_ir.cc50 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()
Dget_compiler_ir.h30 enum class IrExportStage { enum
41 IrExportStage stage, ProcessFunctionLibraryRuntime* pflr,
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc309 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()