Home
last modified time | relevance | path

Searched refs:FunctionalizedRngRuntimeWrapper (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Djit_compile_runtime_wrappers.py52 FunctionalizedRngRuntimeWrapper,
152 functionalized_rng_wrapper = FunctionalizedRngRuntimeWrapper()
574 functionalized_rng_wrapper = FunctionalizedRngRuntimeWrapper(
H A Dautograd_cache.py36 FunctionalizedRngRuntimeWrapper,
359 compiled_fw_func = FunctionalizedRngRuntimeWrapper(
H A Druntime_wrappers.py439 class FunctionalizedRngRuntimeWrapper(CompilerWrapper): class
1910 … out = FunctionalizedRngRuntimeWrapper()._functionalized_rng_runtime_epilogue(
1993 out = FunctionalizedRngRuntimeWrapper()._functionalized_rng_runtime_epilogue(