pytorch
cd7408e9 - Add aten _assert_tensor_metadata op (#84617)

Commit
2 years ago
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
Author
Committer
Parents
Loading