This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
!9081
int32->int64 support
From:
@yepei6
Reviewed-by: @kingxian,@zh_qh Signed-off-by:
@kingxian
tags/v1.1.0
mindspore-ci-bot
Gitee
5 years ago
parent
e211e11b48
9cbb9bc428
commit
cf0bca06eb
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
mindspore/ccsrc/backend/kernel_compiler/gpu/quant/batchnorm_fold2_grad_gpu_kernel.h
+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/quant/batchnorm_fold2_grad_gpu_kernel.h
View File
@@ -114,7 +114,7 @@ class BatchNormFold2GradGpuKernel : public GpuKernel {
channel_ = input_shape[1];
height_ = input_shape[2];
width_ = input_shape[3];
freeze_bn_ = GetValue<int
32
_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("freeze_bn"));
freeze_bn_ = GetValue<int
64
_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("freeze_bn"));
InitSizeLists();
return true;
Write
Preview
Loading…
Cancel
Save