Searched refs:while_ops (Results 1 – 3 of 3) sorted by relevance
218 llvm::SmallVector<WhileOp, 4> while_ops; in runOnFunction() local219 func.walk([&](WhileOp op) { while_ops.push_back(op); }); in runOnFunction()221 for (auto& op : while_ops) { in runOnFunction()
1198 std::vector<HloInstruction*> while_ops; in Run() local1202 while_ops.push_back(instr); in Run()1207 for (HloInstruction* while_op : while_ops) { in Run()
477 while_ops = [op for op in graph.get_operations() if "While" in op.type]478 self.assertAllEqual([op.type for op in while_ops], ["StatelessWhile"] * 4,480 for op in while_ops: