diff options
Diffstat (limited to 'tensorflow/contrib/lite/kernels/strided_slice_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/strided_slice_test.cc | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/tensorflow/contrib/lite/kernels/strided_slice_test.cc b/tensorflow/contrib/lite/kernels/strided_slice_test.cc index 716b11d432..c5d4f9affb 100644 --- a/tensorflow/contrib/lite/kernels/strided_slice_test.cc +++ b/tensorflow/contrib/lite/kernels/strided_slice_test.cc @@ -21,7 +21,6 @@ limitations under the License. namespace tflite { namespace { -using ::int32; using ::testing::ElementsAreArray; template <typename input_type = float, @@ -50,14 +49,14 @@ class StridedSliceOpModel : public SingleOpModel { void SetInput(std::initializer_list<input_type> data) { PopulateTensor<input_type>(input_, data); } - void SetBegin(std::initializer_list<int32> data) { - PopulateTensor<int32>(begin_, data); + void SetBegin(std::initializer_list<int32_t> data) { + PopulateTensor<int32_t>(begin_, data); } - void SetEnd(std::initializer_list<int32> data) { - PopulateTensor<int32>(end_, data); + void SetEnd(std::initializer_list<int32_t> data) { + PopulateTensor<int32_t>(end_, data); } - void SetStrides(std::initializer_list<int32> data) { - PopulateTensor<int32>(strides_, data); + void SetStrides(std::initializer_list<int32_t> data) { + PopulateTensor<int32_t>(strides_, data); } std::vector<input_type> GetOutput() { @@ -566,7 +565,7 @@ TEST(StridedSliceOpTest, RunTwice) { } TEST(StridedSliceOpTest, In3D_IdentityShrinkAxis1Uint8) { - StridedSliceOpModel<uint8, TensorType_UINT8> m({2, 3, 2}, {3}, {3}, {3}, 0, 0, + StridedSliceOpModel<uint8_t, TensorType_UINT8> m({2, 3, 2}, {3}, {3}, {3}, 0, 0, 0, 0, 1); m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); m.SetBegin({0, 0, 0}); |