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

Update tensorflow/core/util/padding.cc

Browse Source 
Co-authored-by: Mihai Maruseac <mihai.maruseac@gmail.com>
This commit is contained in:
tg-at-google 2020-07-28 20:59:14 -04:00 committed by GitHub
parent 3fb90eb802
commit 5d14e877b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/util/padding.cc
View File

@ -37,8 +37,8 @@ Status CheckValidPadding(Padding padding_type,
const std::vector<int64>& explicit_paddings,
int num_dims, TensorFormat data_format) {
if (padding_type == Padding::EXPLICIT) {
const int explicit_paddings_size = explicit_paddings.size();
if (explicit_paddings_size != 2 * num_dims) {
const int num_paddings = explicit_paddings.size();
if (num_paddings != 2 * num_dims) {
return errors::InvalidArgument(
"explicit_paddings attribute must contain ", 2 * num_dims,
" values, but got: ", explicit_paddings.size());