1 /* 2 * Copyright (c) Meta Platforms, Inc. and affiliates. 3 * All rights reserved. 4 * 5 * This source code is licensed under the BSD-style license found in the 6 * LICENSE file in the root directory of this source tree. 7 */ 8 9 #pragma once 10 11 #include <executorch/runtime/kernel/kernel_includes.h> 12 13 namespace torch { 14 namespace executor { 15 16 /* 17 * Repeats in tensor along the dimensions specified in repeats. 18 * 19 * @param[in] Input tensor that we want to repeat. 20 * @param[in] The number of times to repeat this tensor along each dimension 21 * @param[in] Output tensor to write to. 22 * 23 * @returns The status of the repeat operation. 24 */ 25 Error repeat_tensor( 26 const exec_aten::Tensor& in, 27 exec_aten::ArrayRef<int64_t> repeats, 28 exec_aten::Tensor& out); 29 30 } // namespace executor 31 } // namespace torch 32