/* * Copyright (c) Meta Platforms, Inc. and affiliates. * All rights reserved. * * This source code is licensed under the BSD-style license found in the * LICENSE file in the root directory of this source tree. */ #include #include #include namespace torch { namespace executor { namespace native { using Tensor = exec_aten::Tensor; Tensor& select_copy_int_out( KernelRuntimeContext& ctx, const Tensor& in, int64_t dim, int64_t index, Tensor& out) { Error err = torch::executor::select_copy_util(in, dim, index, out); if (err != Error::Ok) { ctx.fail(err); } return out; } } // namespace native } // namespace executor } // namespace torch