From 330cce116848791a1393ce523033c08b102ac792 Mon Sep 17 00:00:00 2001 From: npt-1707 Date: Fri, 9 May 2025 04:08:42 +0800 Subject: [PATCH] tensorflow/lite/kernels: Prevent division by zero in conv_grad_ops.cc Signed-off-by: npt-1707 --- tensorflow/core/kernels/conv_grad_ops.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tensorflow/core/kernels/conv_grad_ops.cc b/tensorflow/core/kernels/conv_grad_ops.cc index 7326dcc58b4..1a11835cc1d 100644 --- a/tensorflow/core/kernels/conv_grad_ops.cc +++ b/tensorflow/core/kernels/conv_grad_ops.cc @@ -131,6 +131,10 @@ Status ConvBackpropComputeDimensionsV2( // dimensions of the filter Tensor. VLOG(2) << "input vs filter_in depth " << dims->in_depth << " " << filter_shape.dim_size(num_dims - 2); + if (filter_shape.dim_size(num_dims - 2) <= 0) { + return errors ::InvalidArgument( + label, ": filter depth must be strictly greated than zero"); + } if (dims->in_depth % filter_shape.dim_size(num_dims - 2)) { return errors::InvalidArgument( label, ": input depth must be evenly divisible by filter depth");