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

Update bcast.h

Browse Source
This commit is contained in:
tg-at-google 2020-07-15 11:17:35 -04:00 committed by GitHub
parent 603104ecf5
commit 16dcd34747
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

2
tensorflow/core/util/bcast.h
View File

@ -176,7 +176,7 @@ BCastList<N>::BCastList(const BCastList::Vec (&x)[N],
// 1-extend and align all vectors. // 1-extend and align all vectors.
for (int i = 0; i < N; ++i) { for (int i = 0; i < N; ++i) {
if (copy_i_size < largest_rank) { if (copy[i].size() < largest_rank) {
copy[i].resize(largest_rank, 1); copy[i].resize(largest_rank, 1);
} }
} }