From ec0fe999ca39f6279e15ba8492a24798a9195fa0 Mon Sep 17 00:00:00 2001 From: Marisa Kirisame Date: Sat, 23 May 2020 09:54:46 -0700 Subject: [PATCH] save --- aten/src/ATen/native/native_functions.yaml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml index 667250ce86e..87cd7cb7165 100644 --- a/aten/src/ATen/native/native_functions.yaml +++ b/aten/src/ATen/native/native_functions.yaml @@ -3,6 +3,9 @@ - func: checkpoint(Tensor self) -> Tensor variants: method +- func: try_checkpoint(Tensor self) -> Tensor + variants: method + - func: is_checkpoint(Tensor self) -> bool variants: method