save

uwsampl · May 23, 2020 · ec0fe99 · ec0fe99
1 parent 73f681c
commit ec0fe99
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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