diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-19 14:27:43 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-19 14:30:45 -0700 |
| commit | 48832eff2833c34294a46d49af5a78c9318ca528 (patch) | |
| tree | 761ad2ab2deb61e5c61a4f413880ff5ce4bf7794 /tensorflow/contrib/lite/kernels/softmax_test.cc | |
| parent | 27c27c58e1f8b4ac86f85eb201f0d9d667fa83a1 (diff) | |
Automated g4 rollback of changelist 201194552
PiperOrigin-RevId: 201241214
Diffstat (limited to 'tensorflow/contrib/lite/kernels/softmax_test.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/softmax_test.cc | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/tensorflow/contrib/lite/kernels/softmax_test.cc b/tensorflow/contrib/lite/kernels/softmax_test.cc index 727822f6be..6c5338ff0f 100644 --- a/tensorflow/contrib/lite/kernels/softmax_test.cc +++ b/tensorflow/contrib/lite/kernels/softmax_test.cc @@ -92,9 +92,10 @@ TEST(SoftmaxOpTest, CompareWithTFminiBetaEq1) { m.Invoke(); 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::Softmax(input_buffer, input_shape, beta, - output_buffer.get(), input_shape); + 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); std::vector<float> expected; expected.insert(expected.end(), output_buffer.get(), @@ -119,9 +120,10 @@ TEST(SoftmaxOpTest, CompareWithTFminiBetaNotEq1) { m.Invoke(); 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::Softmax(input_buffer, input_shape, beta, - output_buffer.get(), input_shape); + 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); std::vector<float> expected; expected.insert(expected.end(), output_buffer.get(), |