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

Merge pull request #40511 from Meteorix:patch-1

Browse Source 
PiperOrigin-RevId: 317056448
Change-Id: Ia160c46af83a372d8d286151e30f06a4e02b1fa3
This commit is contained in:
TensorFlower Gardener 2020-06-18 01:38:40 -07:00
commit f9c4663043
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(