add parameter output_name

PaddlePaddle · Aug 4, 2022 · 0072ec3 · 0072ec3
1 parent 6b65600
commit 0072ec3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/rfcs/CINN/APIs/20220802_cinn_api_design_one_hot.md b/rfcs/CINN/APIs/20220802_cinn_api_design_one_hot.md
@@ -4,7 +4,7 @@
 | -------------- | ----------------------------------- |
 | 提交作者       | Nyakku Shigure（@SigureMo）         |
 | 提交时间       | 2022-08-02                          |
-| 版本号         | v0.4                                |
+| 版本号         | v0.5                                |
 | 依赖 CINN 版本 | develop                             |
 | 文件名         | 20220802_cinn_api_design_one_hot.md |
 
@@ -369,7 +369,8 @@ ir::Tensor OneHot(const ir::Tensor& indices,
                   const ir::Tensor& off_value,
                   const int depth,
                   const int axis,
-                  const Type& dtype);
+                  const Type& dtype,
+                  const std::string& output_name);
 ```
 
 ### 底层 OP 设计