experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix LaunchDepthwiseConvBackpropFilterOp

Browse Source
This commit is contained in:
Meteorix 2020-06-16 23:38:53 +08:00 committed by GitHub
parent cb60e1c14b
commit bedc750c7d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/depthwise_conv_op_gpu.h
View File

@ -1763,7 +1763,7 @@ void LaunchDepthwiseConvBackpropFilterOp<GpuDevice, T>::operator()(
int num_filter_backprop = int num_filter_backprop =
args.filter_rows * args.filter_cols * args.out_depth; args.filter_rows * args.filter_cols * args.out_depth;
se::DeviceMemoryBase filter_bp_ptr(filter_backprop, num_filter_backprop); se::DeviceMemoryBase filter_bp_ptr(filter_backprop, num_filter_backprop);
stream->ThenMemset32(&filter_bp_ptr, 0, num_filter_backprop * sizeof(T)); stream->ThenMemZero(&filter_bp_ptr, num_filter_backprop * sizeof(T));
if (args.filter_rows == 3 && args.filter_cols == 3) { if (args.filter_rows == 3 && args.filter_cols == 3) {
OP_REQUIRES_OK( OP_REQUIRES_OK(