Searched refs:should_optimize (Results 1 – 3 of 3) sorted by relevance
60 name, map_fn, should_optimize = y63 should_optimize=should_optimize)89 def testHoistFunction(self, map_fn, should_optimize): argument92 ["Zip[0]", "Map"] if should_optimize else ["Map"])).map(map_fn)
50 name, function, should_optimize = y53 should_optimize=should_optimize)63 def testMapParallelization(self, function, should_optimize): argument64 next_nodes = ["ParallelMap"] if should_optimize else ["Map"]
41 def should_optimize(): function66 if args.check_server and not should_optimize():