Searched refs:max_fused_ops (Results 1 – 3 of 3) sorted by relevance
581 def expected(n, max_fused_ops): argument584 for i in range(max_fused_ops):593 for i in range(n - max_fused_ops):600 max_fused_ops = 256605 after = run_opt_pass(expected(n, max_fused_ops), transform.InferType())608 max_fused_ops = 10611 after = run_opt_pass(expected(n, max_fused_ops), transform.InferType())613 with tvm.transform.PassContext(config={"relay.FuseOps.max_depth": max_fused_ops}):745 max_fused_ops = branch_len * 2 + 1 # the number of ops in one diamond748 with tvm.transform.PassContext(config={"relay.FuseOps.max_depth": max_fused_ops}):
589 max_fused_ops = 256602 for i in range(max_fused_ops):610 for i in range(n-max_fused_ops):