xref: /aosp_15_r20/external/executorch/extension/llm/custom_ops/op_tile_crop.h (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
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 namespace native {
17 
18 Tensor& tile_crop_out_impl(
19     KernelRuntimeContext& ctx,
20     const Tensor& input,
21     const int64_t tile_size,
22     Tensor& out);
23 
24 } // namespace native
25 } // namespace executor
26 } // namespace torch
27