OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #49954 from tensorflow/cp2_3_369343977

Browse Source 
Prevent yet another division by zero
This commit is contained in:
Mihai Maruseac
2021-06-02 10:06:43 -07:00
committed by GitHub
parent c549f3ea77 38aa2a4310
commit 32eb05b5ef
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/kernels/conv_grad_input_ops.cc
View File

@@ -668,6 +668,11 @@ class Conv2DCustomBackpropInputOp : public OpKernel {
dims.batch_size == 1 ||
thread_work_unit_size >= min_thread_work_unit_size;
OP_REQUIRES(
context, work_unit_size > 0,
errors::InvalidArgument("input, filter_sizes and out_backprop tensors "
"must all have at least 1 element"));
const size_t shard_size =
use_parallel_contraction
? 1