Searched refs:estimate_op_runtime (Results 1 – 2 of 2) sorted by relevance
154 snode_to_cost = {snode: estimate_op_runtime(snode) for snode in snodes}255 def estimate_op_runtime(snode: BaseSchedulerNode) -> float: function259 if config.estimate_op_runtime == "default":262 assert callable(config.estimate_op_runtime)263 runtime = config.estimate_op_runtime(snode)292 total_est_runtime += estimate_op_runtime(snode)299 total_est_runtime += estimate_op_runtime(snode)
267 estimate_op_runtime = "default" variable