diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-29 13:28:40 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-29 13:31:21 -0700 |
| commit | d3c0598ff7650b9d25eb01f792a6106428163df5 (patch) | |
| tree | cd2a2e35cc743a8273bd9b34f019b425dbf85e6d /tensorflow/contrib/lite/kernels/embedding_lookup.cc | |
| parent | f95b37f6d1b5fd3c9ef71d974763c2248f690b54 (diff) | |
Use the same convention for the scale parameter in hybrid ops as well.
PiperOrigin-RevId: 202698287
Diffstat (limited to 'tensorflow/contrib/lite/kernels/embedding_lookup.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/embedding_lookup.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/embedding_lookup.cc b/tensorflow/contrib/lite/kernels/embedding_lookup.cc index 9410bead5e..0ba170a4da 100644 --- a/tensorflow/contrib/lite/kernels/embedding_lookup.cc +++ b/tensorflow/contrib/lite/kernels/embedding_lookup.cc @@ -94,7 +94,7 @@ TfLiteStatus EvalHybrid(TfLiteContext* context, TfLiteNode* node, const TfLiteTensor* lookup, const TfLiteTensor* value, TfLiteTensor* output) { const int row_size = SizeOfDimension(value, 0); - const double scaling_factor = 1.0 / value->params.scale; + const double scaling_factor = value->params.scale; // col_size after we flatten tensor into 2D. int col_size = 1; |