diff --git a/torch_xla/csrc/tensor_methods.h b/torch_xla/csrc/tensor_methods.h index 94bb96a4863..8465ff23269 100644 --- a/torch_xla/csrc/tensor_methods.h +++ b/torch_xla/csrc/tensor_methods.h @@ -752,6 +752,9 @@ XLATensorPtr rsub( void copy_(XLATensorPtr& input, XLATensorPtr& src); +XLATensorPtr scan(const Callable f, const XLATensorPtr& init, + const XLATensorPtr& xs); + XLATensorPtr scatter(const XLATensorPtr& input, int64_t dim, const XLATensorPtr& index, const XLATensorPtr& src); XLATensorPtr scatter(const XLATensorPtr& input, int64_t dim,