Searched refs:evaluate_guards_expression (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/test/ |
| D | test_dynamic_shapes.py | 2551 self.assertTrue(shape_env.evaluate_guards_expression(guards, [hint_int(s0)]))
2552 self.assertFalse(shape_env.evaluate_guards_expression(guards, [hint_int(s1)]))
2553 self.assertFalse(shape_env.evaluate_guards_expression(guards, [hint_int(s2)]))
2560 self.assertTrue(shape_env.evaluate_guards_expression(guards, [hint_int(s0)]))
2572 self.assertTrue(shape_env.evaluate_guards_expression(guards, [hint_int(s0)]))
2573 self.assertFalse(shape_env.evaluate_guards_expression(guards, [hint_int(s1)]))
|
| /external/pytorch/torch/_inductor/ |
| D | codecache.py | 839 shape_env.evaluate_guards_expression(candidate.guards_expr, hints)
890 shape_env.evaluate_guards_expression(graph.guards_expr, symints)
|
| /external/pytorch/torch/fx/experimental/ |
| D | symbolic_shapes.py | 4260 def evaluate_guards_expression(self, code, args): member in ShapeEnv
4273 return self.evaluate_guards_expression(code, args)
|