Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dconfig.py121 activation_memory_budget_runtime_estimator = "flops" variable
H A Dpartitioners.py1499 RUNTIME_MODE = config.activation_memory_budget_runtime_estimator
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_ac.py291 @config.patch(activation_memory_budget_runtime_estimator="profile")