Searched refs:should_optimize (Results 1 – 3 of 3) sorted by relevance
47 name, function, should_optimize = y50 should_optimize=should_optimize)60 def testMapParallelization(self, function, should_optimize): argument61 next_nodes = ["ParallelMap"] if should_optimize else ["Map"]
9 def optimized_execution(should_optimize): argument12 torch._C._set_graph_executor_optimize(should_optimize)
243 const bool should_optimize = dynamic_control_->OptimizeOperation( in PerformSourceCopyOperation() local245 const InstallOperation& optimized = should_optimize ? buf : operation; in PerformSourceCopyOperation()