已合并
修改AdaptiveAvgPool3d proto注释 #2303
第一章创建于 3月4日
修改AdaptiveAvgPool3d proto注释 #2303
已合并
共 1 个文件变更+6-5
| @@ -27,16 +27,17 @@ namespace ge { | |||
| 27 | * One input, including: | 27 | * One input, including: |
| 28 | * @li x: A Tensor. Must be one of the following data types: | 28 | * @li x: A Tensor. Must be one of the following data types: |
| 29 | * float16, bfloat16, float32. \n | 29 | * float16, bfloat16, float32. \n |
| 30 | +* For Ascend 950PR/Ascend 950DT: Support Format: [NDHWC, NCDHW]. | ||
| 30 | 31 | ||
| 31 | * @par Attributes: | 32 | * @par Attributes: |
| 32 | * @li output_size: A required list of 3 ints | 33 | * @li output_size: A required list of 3 ints |
| 33 | -* specifying the size (D,H,W) of the output tensor. \n | 34 | +* specifying the size (D,H,W) of the output tensor. \n |
| 35 | +* @li data_format: An optional string, Specify the data format of the input and | ||
| 36 | +* output data. either "NDHWC"(default), or "NCDHW" . \n | ||
| 37 | +* For Ascend 950PR/Ascend 950DT: Supported "NDHWC", and "NCDHW". | ||
| 34 | 38 | ||
| 35 | * @par Outputs: | 39 | * @par Outputs: |
| 36 | -* @li y: A Tensor. Has the same data type as "x" \n | 40 | +* @li y: A Tensor. Has the same data type and the same format as "x" \n |
| 37 | - | ||
| 38 | -* @li data_format: An optional string, Specify the data format of the input and | ||
| 39 | -* output data. With the default format "NDHWC" . \n | ||
| 40 | 41 | ||
| 41 | * @par Third-party framework compatibility | 42 | * @par Third-party framework compatibility |
| 42 | * Compatible with the Pytorch operator AdaptiveAvgPool3d. | 43 | * Compatible with the Pytorch operator AdaptiveAvgPool3d. |