Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
Paddle
提交
5df464fe
P
Paddle
项目概览
BaiXuePrincess
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
5df464fe
编写于
8月 25, 2022
作者:
W
wanghuancoder
提交者:
GitHub
8月 25, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[Eager] sync_batch_norm_grad delete mean and variance (#45411)
* sync_batch_norm_grad delete mean and variance
上级
1cd7e68b
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
2 addition
and
8 deletion
+2
-8
paddle/phi/api/yaml/legacy_backward.yaml
paddle/phi/api/yaml/legacy_backward.yaml
+2
-2
paddle/phi/kernels/gpu/sync_batch_norm_grad_kernel.cu
paddle/phi/kernels/gpu/sync_batch_norm_grad_kernel.cu
+0
-2
paddle/phi/kernels/sync_batch_norm_grad_kernel.h
paddle/phi/kernels/sync_batch_norm_grad_kernel.h
+0
-2
paddle/phi/ops/compat/sync_batch_norm_sig.cc
paddle/phi/ops/compat/sync_batch_norm_sig.cc
+0
-2
未找到文件。
paddle/phi/api/yaml/legacy_backward.yaml
浏览文件 @
5df464fe
...
...
@@ -2460,7 +2460,7 @@
-
backward_api
:
sync_batch_norm_grad
forward
:
sync_batch_norm_ (Tensor x, Tensor scale, Tensor bias, Tensor mean, Tensor variance, float momentum, float epsilon, str data_layout, bool is_test, bool use_global_stats, bool trainable_statistics, bool fuse_with_relu) -> Tensor(out), Tensor(mean_out), Tensor(variance_out), Tensor(saved_mean), Tensor(saved_variance), Tensor(reserve_space)
args
:
(Tensor x, Tensor scale, Tensor bias, Tensor
mean_out, Tensor variance_out, Tensor
saved_mean, Tensor saved_variance, Tensor reserve_space, Tensor out_grad, float momentum, float epsilon, str data_layout, bool is_test, bool use_global_stats, bool trainable_statistics, bool fuse_with_relu)
args
:
(Tensor x, Tensor scale, Tensor bias, Tensor saved_mean, Tensor saved_variance, Tensor reserve_space, Tensor out_grad, float momentum, float epsilon, str data_layout, bool is_test, bool use_global_stats, bool trainable_statistics, bool fuse_with_relu)
output
:
Tensor(x_grad), Tensor(scale_grad), Tensor(bias_grad)
infer_meta
:
func
:
GeneralTernaryGradInferMeta
...
...
@@ -2468,7 +2468,7 @@
kernel
:
func
:
sync_batch_norm_grad
data_type
:
out_grad
optional
:
mean_out, variance_out,
reserve_space
optional
:
reserve_space
-
backward_api
:
take_along_axis_grad
forward
:
take_along_axis (Tensor x, Tensor index, int axis) -> Tensor(out)
...
...
paddle/phi/kernels/gpu/sync_batch_norm_grad_kernel.cu
浏览文件 @
5df464fe
...
...
@@ -24,8 +24,6 @@ void SyncBatchNormGradKernel(const Context& ctx,
const
DenseTensor
&
x
,
const
DenseTensor
&
scale
,
const
DenseTensor
&
bias
,
const
paddle
::
optional
<
DenseTensor
>&
mean
,
const
paddle
::
optional
<
DenseTensor
>&
variance
,
const
DenseTensor
&
saved_mean
,
const
DenseTensor
&
saved_variance
,
const
paddle
::
optional
<
DenseTensor
>&
reserve_space
,
...
...
paddle/phi/kernels/sync_batch_norm_grad_kernel.h
浏览文件 @
5df464fe
...
...
@@ -25,8 +25,6 @@ void SyncBatchNormGradKernel(const Context& dev_ctx,
const
DenseTensor
&
x
,
const
DenseTensor
&
scale
,
const
DenseTensor
&
bias
,
const
paddle
::
optional
<
DenseTensor
>&
mean
,
const
paddle
::
optional
<
DenseTensor
>&
variance
,
const
DenseTensor
&
saved_mean
,
const
DenseTensor
&
saved_variance
,
const
paddle
::
optional
<
DenseTensor
>&
reserve_space
,
...
...
paddle/phi/ops/compat/sync_batch_norm_sig.cc
浏览文件 @
5df464fe
...
...
@@ -42,8 +42,6 @@ KernelSignature SyncBatchNormGradOpArgumentMapping(
"X"
,
"Scale"
,
"Bias"
,
"Mean"
,
"Variance"
,
"SavedMean"
,
"SavedVariance"
,
"ReserveSpace"
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录