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/softmax_test.cc | |
parent | 1a517b99b6c2c1abbe5390f87f4128db5e69e142 (diff) |
Automated g4 rollback of changelist 201241214
PiperOrigin-RevId: 201410380
Diffstat (limited to 'tensorflow/contrib/lite/kernels/softmax_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/softmax_test.cc | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/tensorflow/contrib/lite/kernels/softmax_test.cc b/tensorflow/contrib/lite/kernels/softmax_test.cc index 6c5338ff0f..727822f6be 100644 --- a/tensorflow/contrib/lite/kernels/softmax_test.cc +++ b/tensorflow/contrib/lite/kernels/softmax_test.cc @@ -92,10 +92,9 @@ TEST(SoftmaxOpTest, CompareWithTFminiBetaEq1) { 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::Softmax(input_buffer, input_dims, beta, - output_buffer.get(), input_dims); + auto input_shape = RuntimeShape({batch_size, 1, 1, input_size}); + tflite::reference_ops::Softmax(input_buffer, input_shape, beta, + output_buffer.get(), input_shape); std::vector<float> expected; expected.insert(expected.end(), output_buffer.get(), @@ -120,10 +119,9 @@ TEST(SoftmaxOpTest, CompareWithTFminiBetaNotEq1) { 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::Softmax(input_buffer, input_dims, beta, - output_buffer.get(), input_dims); + auto input_shape = RuntimeShape({batch_size, 1, 1, input_size}); + tflite::reference_ops::Softmax(input_buffer, input_shape, beta, + output_buffer.get(), input_shape); std::vector<float> expected; expected.insert(expected.end(), output_buffer.get(), |