Searched refs:LoopStateTy (Results 1 – 2 of 2) sorted by relevance
62 using LoopStateTy = std::vector<HloInstruction*>; variable63 using LoopBodyGeneratorTy = std::function<StatusOr<LoopStateTy>(65 const LoopStateTy& /*current_values*/)>;79 static StatusOr<LoopStateTy> MakeCountedLoop(81 const LoopStateTy& init_values,
176 const std::function<StatusOr<WhileUtil::LoopStateTy>( in MakeCountedLoopBodyComputation()177 HloInstruction*, const WhileUtil::LoopStateTy&)>& loop_body_generator) { in MakeCountedLoopBodyComputation()206 HloComputation* computation, const WhileUtil::LoopStateTy& init_values) { in MakeInitTupleFromInitValues()217 static Shape MakeLoopStateShape(const WhileUtil::LoopStateTy& init_values) { in MakeLoopStateShape()227 /*static*/ StatusOr<WhileUtil::LoopStateTy> WhileUtil::MakeCountedLoop( in MakeCountedLoop()229 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop()