OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Make IValue::unsafeToTensorImpl a little less unsafe. (#88043)

Browse Source 
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88043
Approved by: https://github.com/anjali411, https://github.com/albanD
This commit is contained in:
Edward Z. Yang 2022-10-29 21:43:09 -04:00 committed by PyTorch MergeBot
parent e238752e20
commit 8ea19c802e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
aten/src/ATen/core/ivalue.h
View File

@ -425,6 +425,7 @@ public:
at::Tensor& toTensor() &; at::Tensor& toTensor() &;
const at::Tensor& toTensor() const&; const at::Tensor& toTensor() const&;
at::TensorImpl* unsafeToTensorImpl() const { at::TensorImpl* unsafeToTensorImpl() const {
TORCH_INTERNAL_ASSERT(isTensor());
return payload.as_tensor.unsafeGetTensorImpl(); return payload.as_tensor.unsafeGetTensorImpl();
} }