diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-11 14:10:40 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-11 14:13:27 -0700 |
commit | 0d9b4f06b7242288a3aeb0d29fe10278522c7f45 (patch) | |
tree | 3f7ef4e88304eb285b2c0c3cfebd3c25b72d3058 /tensorflow/contrib/lite/kernels/fully_connected.cc | |
parent | 3be426254eb8f0066deb0324c5237786045245c1 (diff) |
Internal Change.
PiperOrigin-RevId: 200109989
Diffstat (limited to 'tensorflow/contrib/lite/kernels/fully_connected.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/fully_connected.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/fully_connected.cc b/tensorflow/contrib/lite/kernels/fully_connected.cc index 989920622d..5a0524bec6 100644 --- a/tensorflow/contrib/lite/kernels/fully_connected.cc +++ b/tensorflow/contrib/lite/kernels/fully_connected.cc @@ -105,7 +105,7 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { const int batch_size = input_size / filter->dims->data[1]; const int num_units = filter->dims->data[0]; - TF_LITE_ASSERT_EQ(input_size, batch_size * filter->dims->data[1]); + TF_LITE_ENSURE_EQ(context, input_size, batch_size * filter->dims->data[1]); if (bias) { TF_LITE_ENSURE_EQ(context, NumElements(bias), SizeOfDimension(filter, 0)); } |