Searched refs:_insert_assert_async (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/_export/passes/ |
| D | add_runtime_assertions_for_constraints_pass.py | 58 last_node = self._insert_assert_async(last_node, operator.ge, node, lower, assert_msg)
61 last_node = self._insert_assert_async(last_node, operator.le, node, upper, assert_msg)
63 def _insert_assert_async(self, last_node, op, lower, upper, assert_msg): member in _AddRuntimeAssertionsForInlineConstraintsPass
|