Home
last modified time | relevance | path

Searched defs:leaf_guard (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dguards.cpp1648 virtual void add_leaf_guard(std::shared_ptr<LeafGuard> leaf_guard) { in add_leaf_guard()
1831 void add_permitted_leaf_guard(std::shared_ptr<LeafGuard> leaf_guard) { in add_permitted_leaf_guard()
1975 void add_epilogue_lambda_guard(std::unique_ptr<LeafGuard> leaf_guard) { in add_epilogue_lambda_guard()
2246 void add_leaf_guard(std::shared_ptr<LeafGuard> leaf_guard) override { in add_leaf_guard()
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dguards.py268 def get_code_parts(leaf_guard): argument