diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-20 13:55:35 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-20 13:57:53 -0700 |
commit | 35616039860ab25dde6f87b9a9e87f8727fa0daf (patch) | |
tree | a9aa56e20e34128f44e5a4dcae3f299972a6729b /tensorflow/contrib/lite/kernels/log_softmax_test.cc | |
parent | 1a517b99b6c2c1abbe5390f87f4128db5e69e142 (diff) |
Automated g4 rollback of changelist 201241214
PiperOrigin-RevId: 201410380
Diffstat (limited to 'tensorflow/contrib/lite/kernels/log_softmax_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/log_softmax_test.cc | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/kernels/log_softmax_test.cc b/tensorflow/contrib/lite/kernels/log_softmax_test.cc index 62820a2f51..9a8d35e82c 100644 --- a/tensorflow/contrib/lite/kernels/log_softmax_test.cc +++ b/tensorflow/contrib/lite/kernels/log_softmax_test.cc @@ -90,10 +90,9 @@ TEST(LogSoftmaxOpTest, CompareWithTFmini) { m.Invoke(); std::unique_ptr<float[]> output_buffer(new float[input_size * batch_size]); - static tflite::Dims<4> input_dims = {{input_size, 1, 1, batch_size}, - {1, 0, 0, input_size}}; - tflite::reference_ops::LogSoftmax(input_buffer, input_dims, - output_buffer.get(), input_dims); + auto input_shape = RuntimeShape({batch_size, 1, 1, input_size}); + tflite::reference_ops::LogSoftmax(input_buffer, input_shape, + output_buffer.get(), input_shape); std::vector<float> expected; expected.insert(expected.end(), output_buffer.get(), |