Searched refs:get_isolated_graphmodule (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_proxy_tensor.py | 25 …rch.fx.experimental.proxy_tensor import make_fx, DecompositionInterpreter, get_isolated_graphmodule
228 gm = get_isolated_graphmodule(inner, (x,), {})
245 gm = get_isolated_graphmodule(inner_with_factory, (), {})
250 gm = get_isolated_graphmodule(f1, (x,), {})
292 gm = get_isolated_graphmodule(inner_with_factory, (), {})
298 gm = get_isolated_graphmodule(f1_logging, (x,), {})
314 gm = get_isolated_graphmodule(inner_with_factory, (), {})
320 gm = get_isolated_graphmodule(f1_logging_tensor, (x,), {})
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | proxy_tensor.py | 2178 def get_isolated_graphmodule( function
|