Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!49730 [doc]add the GPU platform for avg_pool3d 

Merge pull request !49730 from hanhuifeng/code_docs_avg_pool3d
This commit is contained in:
i-robot 2023-03-07 03:41:53 +00:00 committed by Gitee
parent 4ce984a640 c98e39e365
commit 9bec955a7d
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/nn/layer/pooling.py
View File

@ -543,7 +543,7 @@ class AvgPool3d(Cell):
ValueError: If length of shape of `x` is not equal to 5.
Supported Platforms:
``Ascend`` ``CPU``
``Ascend`` ``GPU`` ``CPU``
Examples:
>>> import mindspore as ms

2
mindspore/python/mindspore/ops/function/nn_func.py
View File

@ -516,7 +516,7 @@ def avg_pool3d(input_x, kernel_size=1, stride=1, padding=0, ceil_mode=False, cou
ValueError: If element of `padding` is less than `0`.
Supported Platforms:
``Ascend`` ``CPU``
``Ascend`` ``GPU`` ``CPU``
Examples:
>>> input_x = Tensor(np.arange(1 * 2 * 2 * 2 * 3).reshape((1, 2, 2, 2, 3)), mindspore.float16)