Michael Voznesensky
|
cd7408e950
|
Add aten _assert_tensor_metadata op (#84617)
Example:
```
graph():
%arg0 : [#users=3] = placeholder[target=arg0]
%arg_guard_equality_check : [#users=1] = call_function[target=torch._tensor_equal](args = (%arg0, (1, 1, 2), (2, 2, 1), torch.float32), kwargs = {})
%_assert_true : [#users=0] = call_function[target=torch._assert_true](args = (%arg_guard_equality_check, Guard evaluation failed equality check for arg0), kwargs = {})
%add : [#users=1] = call_function[target=operator.add](args = (%arg0, 1), kwargs = {})
return ([arg0, arg0], (add, add))
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84617
Approved by: https://github.com/jansel
|
2022-09-19 20:48:09 +00:00 |
|