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

toco: Fix missing check for buffer in ResizeBilinear.

Browse Source 
PiperOrigin-RevId: 181209975
This commit is contained in:
A. Unique TensorFlower 2018-01-08 13:29:40 -08:00 committed by TensorFlower Gardener
parent 314d400ad8
commit 2019820130
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc
View File

@ -631,6 +631,9 @@ void ProcessResizeBilinearOperator(Model* model, ResizeBilinearOperator* op) {
const auto& output_size_shape = output_size_array.shape(); const auto& output_size_shape = output_size_array.shape();
CHECK_EQ(output_size_shape.dimensions_count(), 1); CHECK_EQ(output_size_shape.dimensions_count(), 1);
CHECK_EQ(output_size_shape.dims(0), 2); CHECK_EQ(output_size_shape.dims(0), 2);
if (!output_size_array.buffer) {
return;
}
std::vector<int32> output_shape = std::vector<int32> output_shape =
output_size_array.GetBuffer<ArrayDataType::kInt32>().data; output_size_array.GetBuffer<ArrayDataType::kInt32>().data;
model->arrays[op->outputs[0]]->copy_shape( model->arrays[op->outputs[0]]->copy_shape(