diff options
author | Alan Chiao <alanchiao@google.com> | 2018-07-12 17:55:49 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-12 17:59:36 -0700 |
commit | 6468c599d4b3b5c6fa94d171c1ba6b4254286c23 (patch) | |
tree | 329833d0916b31f4f27ba5df8284292e915b472e /tensorflow/contrib/lite/kernels/embedding_lookup.cc | |
parent | fb9992423c20ed716124f63b9fe7858a0198e897 (diff) |
Internal change.
PiperOrigin-RevId: 204398429
Diffstat (limited to 'tensorflow/contrib/lite/kernels/embedding_lookup.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/embedding_lookup.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/embedding_lookup.cc b/tensorflow/contrib/lite/kernels/embedding_lookup.cc index 0ba170a4da..f550339d03 100644 --- a/tensorflow/contrib/lite/kernels/embedding_lookup.cc +++ b/tensorflow/contrib/lite/kernels/embedding_lookup.cc @@ -112,8 +112,9 @@ TfLiteStatus EvalHybrid(TfLiteContext* context, TfLiteNode* node, // TODO(alanchiao): refactor scalar multiply into separate function // for ease of adding a neon equivalent if ever necessary. for (int j = 0; j < col_size; j++) { + const int8_t* value_ptr = reinterpret_cast<int8_t*>(value->data.uint8); output->data.f[j + i * col_size] = - value->data.uint8[j + idx * col_size] * scaling_factor; + value_ptr[j + idx * col_size] * scaling_factor; } } } |