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

Fix a bug where vector::reserve was being called on config.sparse instead of config.ragged.

Browse Source 
PiperOrigin-RevId: 337133816
Change-Id: I4b712c2e141201f21f19fb92627981c53bb4280d
This commit is contained in:
A. Unique TensorFlower 2020-10-14 11:30:17 -07:00 committed by TensorFlower Gardener
parent 9eaac42b60
commit c346633dd4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/example_parsing_ops.cc
View File

@ -222,7 +222,7 @@ class ParseExampleOp : public OpKernel {
for (int d = 0; d < attrs_.num_sparse; ++d) {
config.sparse.emplace_back(sparse_keys_t[d], attrs_.sparse_types[d]);
}
config.sparse.reserve(attrs_.num_ragged);
config.ragged.reserve(attrs_.num_ragged);
for (int d = 0; d < attrs_.num_ragged; ++d) {
config.ragged.emplace_back(ragged_keys_t[d], attrs_.ragged_value_types[d],
attrs_.ragged_split_types[d]);