From 5e42369e9c12e765066bbdffbd3a776a6cc679df Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Wed, 17 Jan 2024 14:23:02 -0800
Subject: [PATCH] Update tensor_methods.h

---
 torch_xla/csrc/tensor_methods.h | 3 +++
 1 file changed, 3 insertions(+)

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,