Searched refs:activation_memory_budget_runtime_estimator (Results 1 – 3 of 3) sorted by relevance
121 activation_memory_budget_runtime_estimator = "flops" variable
1499 RUNTIME_MODE = config.activation_memory_budget_runtime_estimator
291 @config.patch(activation_memory_budget_runtime_estimator="profile")