Searched refs:split_const_subgraphs (Results 1 – 3 of 3) sorted by relevance
75 mod_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(mod)121 mod_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(mod)149 mod_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(mod)179 mod_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(mod)226 mod_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(mod)267 mod_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(mod)318 mod_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(mod)352 mod_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(mod)389 gm_folded: const_fold.FoldedGraphModule = const_fold.split_const_subgraphs(gm)427 gm_folded = const_fold.split_const_subgraphs(mod)[all …]
143 def split_const_subgraphs( function
1790 const_folded_jvp_graph = lazy_dynamo_disallow(const_fold.split_const_subgraphs)(