Home
last modified time | relevance | path

Searched defs:insert_point_guard (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dloop_unrolling.cpp61 WithInsertPoint insert_point_guard{loop}; in inlineBody() local
123 WithInsertPoint insert_point_guard(dest); in repeatBody() local
189 WithInsertPoint insert_point_guard{loop}; in unroll() local
H A Donnx.cpp470 WithInsertPoint insert_point_guard(new_block); in NodeToONNX() local
572 WithInsertPoint insert_point_guard(new_block); in NodeToONNX() local
/aosp_15_r20/external/pytorch/torch/jit/
H A D_ir_utils.py25 def insert_point_guard(self, insert_point: Union[torch._C.Node, torch._C.Block]): function