diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-27 06:54:10 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 06:59:45 -0700 |
commit | 3d30dd424c0404ea5349c0d2acdde2acd4e0aa97 (patch) | |
tree | 37e6bd9c7c643d139a314b3df41e29ea4aa27f4d /tensorflow/contrib/lite/kernels/log_softmax_test.cc | |
parent | 234229b014cb0cfe4bf8e9466db79d596085faba (diff) |
Update kernel evals to use new kernel signatures.
PiperOrigin-RevId: 214767788
Diffstat (limited to 'tensorflow/contrib/lite/kernels/log_softmax_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/log_softmax_test.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/log_softmax_test.cc b/tensorflow/contrib/lite/kernels/log_softmax_test.cc index 9a8d35e82c..1acc966cdc 100644 --- a/tensorflow/contrib/lite/kernels/log_softmax_test.cc +++ b/tensorflow/contrib/lite/kernels/log_softmax_test.cc @@ -91,8 +91,9 @@ TEST(LogSoftmaxOpTest, CompareWithTFmini) { std::unique_ptr<float[]> output_buffer(new float[input_size * batch_size]); auto input_shape = RuntimeShape({batch_size, 1, 1, input_size}); - tflite::reference_ops::LogSoftmax(input_buffer, input_shape, - output_buffer.get(), input_shape); + SoftmaxParams params; + tflite::reference_ops::LogSoftmax(params, input_shape, input_buffer, + input_shape, output_buffer.get()); std::vector<float> expected; expected.insert(expected.end(), output_buffer.get(), |