Searched refs:s_stage (Results 1 – 1 of 1) sorted by relevance
308 std::string s_stage(stage); in InputTFE_OutputTensorHandles() local310 if (s_stage == "hlo") { in InputTFE_OutputTensorHandles()312 } else if (s_stage == "hlo_serialized") { in InputTFE_OutputTensorHandles()314 } else if (s_stage == "optimized_hlo") { in InputTFE_OutputTensorHandles()316 } else if (s_stage == "optimized_hlo_serialized") { in InputTFE_OutputTensorHandles()318 } else if (s_stage == "optimized_hlo_dot") { in InputTFE_OutputTensorHandles()324 s_stage) in InputTFE_OutputTensorHandles()