已合并
feat: 添加LeftShift算子TensorFlow框架插件 #3957
songkai111创建于 7月9日
feat: 添加LeftShift算子TensorFlow框架插件 #3957
已合并
共 1 个文件变更+19-0
| @@ -0,0 +1,19 @@ | |||
| 1 | +/** | ||
| 2 | + * Copyright (c) 2026 Huawei Technologies Co., Ltd. | ||
| 3 | + * This program is free software, you can redistribute it and/or modify it under the terms and conditions of | ||
| 4 | + * CANN Open Software License Agreement Version 2.0 (the "License"). | ||
| 5 | + * Please refer to the License for details. You may not use this file except in compliance with the License. | ||
| 6 | + * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, | ||
| 7 | + * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE. | ||
| 8 | + * See LICENSE in the root of the software repository for the full text of the License. | ||
| 9 | + */ | ||
| 10 | + | ||
| 11 | +#include "register/register.h" | ||
| 12 | + | ||
| 13 | +namespace domi { | ||
| 14 | +REGISTER_CUSTOM_OP("LeftShift") | ||
| 15 | + .FrameworkType(TENSORFLOW) | ||
| 16 | + .OriginOpType("LeftShift") | ||
| 17 | + .ParseParamsByOperatorFn(AutoMappingByOpFn) | ||
| 18 | + .ImplyType(ImplyType::AI_CPU); | ||
| 19 | +} // namespace domi | ||