/**
 * @file function.h
 *
 * Copyright (C) 2024-2025. Huawei Technologies Co., Ltd. All rights reserved.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 */
#ifndef FUNCTION_H
#define FUNCTION_H

#include <ATen/ATen.h>

at::Tensor add_custom_autograd(const at::Tensor& self, const at::Tensor& other);

#endif //  FUNCTION_H