diff --git a/tensorflow/core/kernels/depthwise_conv_op_gpu.h b/tensorflow/core/kernels/depthwise_conv_op_gpu.h index f712d9c48c0..f69878bea89 100644 --- a/tensorflow/core/kernels/depthwise_conv_op_gpu.h +++ b/tensorflow/core/kernels/depthwise_conv_op_gpu.h @@ -1763,7 +1763,7 @@ void LaunchDepthwiseConvBackpropFilterOp::operator()( int num_filter_backprop = args.filter_rows * args.filter_cols * args.out_depth; 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) { OP_REQUIRES_OK(