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

Prevent yet another division by zero

Browse Source
This commit is contained in:
Geeta Chavan 2021-06-01 12:26:15 -07:00
parent b6e130be09
commit 38aa2a4310
1 changed files with 5 additions and 0 deletions
Show Stats 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 || dims.batch_size == 1 ||
thread_work_unit_size >= min_thread_work_unit_size; 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 = const size_t shard_size =
use_parallel_contraction use_parallel_contraction
? 1 ? 1