diff --git a/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc b/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc index 2ca6b251e3a..8f181e78d8e 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc @@ -631,6 +631,9 @@ void ProcessResizeBilinearOperator(Model* model, ResizeBilinearOperator* op) { const auto& output_size_shape = output_size_array.shape(); CHECK_EQ(output_size_shape.dimensions_count(), 1); CHECK_EQ(output_size_shape.dims(0), 2); + if (!output_size_array.buffer) { + return; + } std::vector output_shape = output_size_array.GetBuffer().data; model->arrays[op->outputs[0]]->copy_shape(