diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-25 22:47:21 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-25 22:50:19 -0700 |
commit | 5968eb87766d5536feefaaf8a9ee2f30aeadb354 (patch) | |
tree | a2873e81f4a92a9c3a00e1a89d0ee227c4881c84 /tensorflow/contrib/lite/toco/import_tensorflow.cc | |
parent | 0a01c1451d5353d43c21061261c0fa879f68ad2b (diff) |
Quick fix for import bool array.
PiperOrigin-RevId: 202077590
Diffstat (limited to 'tensorflow/contrib/lite/toco/import_tensorflow.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/import_tensorflow.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/import_tensorflow.cc b/tensorflow/contrib/lite/toco/import_tensorflow.cc index da7e5add7e..485e853e25 100644 --- a/tensorflow/contrib/lite/toco/import_tensorflow.cc +++ b/tensorflow/contrib/lite/toco/import_tensorflow.cc @@ -378,7 +378,7 @@ tensorflow::Status ImportBoolArray(const TensorProto& input_tensor, for (int i = 0; i < input_flat_size; i++) { output_bool_data[i] = input_tensor.bool_val(0); } - } else if (input_tensor.int_val_size() == input_flat_size) { + } else if (input_tensor.bool_val_size() == input_flat_size) { for (int i = 0; i < input_tensor.bool_val_size(); i++) { output_bool_data[i] = input_tensor.bool_val(i); } |