Searched defs:use_orig_params (Results 1 – 4 of 4) sorted by relevance
59 use_orig_params = ["enable_use_orig_params", None] variable
244 self, cpu_offload, sharding_strategy, use_orig_params argument
1917 def _run_test(use_orig_params, optimizer_has_tensor_state): argument
520 use_orig_params=False, argument