Searched refs:VisitAndMutate (Results 1 – 6 of 6) sorted by relevance
347 Stmt VisitAndMutate(Stmt stmt) { in VisitAndMutate() function in tvm::tir::IfThenElseHoister402 return IfThenElseHoister(support_block_scope_hosting).VisitAndMutate(stmt); in HoistIfThenElse()404 Stmt HoistIfThenElse(Stmt stmt) { return IfThenElseHoister().VisitAndMutate(stmt); } in HoistIfThenElse()
340 Stmt VisitAndMutate(Stmt stmt) { in VisitAndMutate() function in tvm::tir::LoopPartitioner576 mid_stmt = VisitAndMutate(mid_stmt); in TryPartition()578 pre_stmt = VisitAndMutate(pre_stmt); in TryPartition()581 post_stmt = VisitAndMutate(post_stmt); in TryPartition()621 stmt = LoopPartitioner(partition_const_loop).VisitAndMutate(std::move(stmt)); in LoopPartition()
314 Stmt VisitAndMutate(const Stmt& stmt) { in VisitAndMutate() function in tvm::ir::LoopPartitioner574 mid_stmt = VisitAndMutate(mid_stmt); in TryPartition()576 pre_stmt = VisitAndMutate(pre_stmt); in TryPartition()579 post_stmt = VisitAndMutate(post_stmt); in TryPartition()622 stmt = LoopPartitioner(split_const_loop).VisitAndMutate(stmt); in LoopPartition()
97 Stmt VisitAndMutate(const Stmt& stmt) { in VisitAndMutate() function in tvm::ir::IfThenElseHoist419 return IfThenElseHoist().VisitAndMutate(stmt); in HoistIfThenElse()