experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #26654 from Dayananda-V:tflite_kernels_error_fix

Browse Source 
PiperOrigin-RevId: 238329955
This commit is contained in:
TensorFlower Gardener 2019-03-13 16:19:50 -07:00
commit 1cf7e422a7
2 changed files with 30 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
tensorflow/lite/kernels/resize_bilinear_test.cc
View File

@ -64,7 +64,7 @@ class ResizeBilinearOpModel : public SingleOpModel {
}; };
TEST(ResizeBilinearOpTest, HorizontalResize) { TEST(ResizeBilinearOpTest, HorizontalResize) {
ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 1, 2, 1}}); ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 1, 2, 1}}, {});
m.SetInput<float>({3, 6}); m.SetInput<float>({3, 6});
m.SetSize({1, 3}); m.SetSize({1, 3});
m.Invoke(); m.Invoke();
@ -79,7 +79,7 @@ TEST(ResizeBilinearOpTest, HorizontalResize) {
} }
TEST(ResizeBilinearOpTest, HorizontalResizeUInt8) { TEST(ResizeBilinearOpTest, HorizontalResizeUInt8) {
ResizeBilinearOpModel m({TensorType_UINT8, {1, 1, 2, 1}}); ResizeBilinearOpModel m({TensorType_UINT8, {1, 1, 2, 1}}, {});
m.SetInput<uint8>({3, 6}); m.SetInput<uint8>({3, 6});
m.SetSize({1, 3}); m.SetSize({1, 3});
m.Invoke(); m.Invoke();
@ -94,7 +94,7 @@ TEST(ResizeBilinearOpTest, HorizontalResizeUInt8) {
} }
TEST(ResizeBilinearOpTest, HorizontalResizeInt8) { TEST(ResizeBilinearOpTest, HorizontalResizeInt8) {
ResizeBilinearOpModel m({TensorType_INT8, {1, 1, 2, 1}}); ResizeBilinearOpModel m({TensorType_INT8, {1, 1, 2, 1}}, {});
m.SetInput<int8_t>({3, 6}); m.SetInput<int8_t>({3, 6});
m.SetSize({1, 3}); m.SetSize({1, 3});
m.Invoke(); m.Invoke();
@ -109,7 +109,7 @@ TEST(ResizeBilinearOpTest, HorizontalResizeInt8) {
} }
TEST(ResizeBilinearOpTest, VerticalResize) { TEST(ResizeBilinearOpTest, VerticalResize) {
ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 1, 1}}); ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 1, 1}}, {});
m.SetInput<float>({3, 9}); m.SetInput<float>({3, 9});
m.SetSize({3, 1}); m.SetSize({3, 1});
m.Invoke(); m.Invoke();
@ -124,7 +124,7 @@ TEST(ResizeBilinearOpTest, VerticalResize) {
} }
TEST(ResizeBilinearOpTest, VerticalResizeUInt8) { TEST(ResizeBilinearOpTest, VerticalResizeUInt8) {
ResizeBilinearOpModel m({TensorType_UINT8, {1, 2, 1, 1}}); ResizeBilinearOpModel m({TensorType_UINT8, {1, 2, 1, 1}}, {});
m.SetInput<uint8>({3, 9}); m.SetInput<uint8>({3, 9});
m.SetSize({3, 1}); m.SetSize({3, 1});
m.Invoke(); m.Invoke();
@ -139,7 +139,7 @@ TEST(ResizeBilinearOpTest, VerticalResizeUInt8) {
} }
TEST(ResizeBilinearOpTest, VerticalResizeInt8) { TEST(ResizeBilinearOpTest, VerticalResizeInt8) {
ResizeBilinearOpModel m({TensorType_INT8, {1, 2, 1, 1}}); ResizeBilinearOpModel m({TensorType_INT8, {1, 2, 1, 1}}, {});
m.SetInput<int8_t>({3, 9}); m.SetInput<int8_t>({3, 9});
m.SetSize({3, 1}); m.SetSize({3, 1});
m.Invoke(); m.Invoke();
@ -154,7 +154,7 @@ TEST(ResizeBilinearOpTest, VerticalResizeInt8) {
} }
TEST(ResizeBilinearOpTest, TwoDimensionalResize) { TEST(ResizeBilinearOpTest, TwoDimensionalResize) {
ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}); ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, {});
m.SetInput<float>({ m.SetInput<float>({
3, 6, // 3, 6, //
9, 12 // 9, 12 //
@ -181,7 +181,7 @@ TEST(ResizeBilinearOpTest, TwoDimensionalResize) {
} }
TEST(ResizeBilinearOpTest, TwoDimensionalResizeUInt8) { TEST(ResizeBilinearOpTest, TwoDimensionalResizeUInt8) {
ResizeBilinearOpModel m({TensorType_UINT8, {1, 2, 2, 1}}); ResizeBilinearOpModel m({TensorType_UINT8, {1, 2, 2, 1}}, {});
m.SetInput<uint8>({ m.SetInput<uint8>({
3, 6, // 3, 6, //
9, 12 // 9, 12 //
@ -208,7 +208,7 @@ TEST(ResizeBilinearOpTest, TwoDimensionalResizeUInt8) {
} }
TEST(ResizeBilinearOpTest, TwoDimensionalResizeInt8) { TEST(ResizeBilinearOpTest, TwoDimensionalResizeInt8) {
ResizeBilinearOpModel m({TensorType_INT8, {1, 2, 2, 1}}); ResizeBilinearOpModel m({TensorType_INT8, {1, 2, 2, 1}}, {});
m.SetInput<int8_t>({ m.SetInput<int8_t>({
3, 6, // 3, 6, //
9, 12 // 9, 12 //
@ -235,7 +235,7 @@ TEST(ResizeBilinearOpTest, TwoDimensionalResizeInt8) {
} }
TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatches) { TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatches) {
ResizeBilinearOpModel m({TensorType_FLOAT32, {2, 2, 2, 1}}); ResizeBilinearOpModel m({TensorType_FLOAT32, {2, 2, 2, 1}}, {});
m.SetInput<float>({ m.SetInput<float>({
3, 6, // 3, 6, //
9, 12, // 9, 12, //
@ -272,7 +272,7 @@ TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatches) {
} }
TEST(ResizeBilinearOpTest, ThreeDimensionalResize) { TEST(ResizeBilinearOpTest, ThreeDimensionalResize) {
ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 2, 2}}); ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 2, 2}}, {});
m.SetInput<float>({ m.SetInput<float>({
3, 4, 6, 10, // 3, 4, 6, 10, //
9, 10, 12, 16, // 9, 10, 12, 16, //
@ -299,7 +299,7 @@ TEST(ResizeBilinearOpTest, ThreeDimensionalResize) {
} }
TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatchesUInt8) { TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatchesUInt8) {
ResizeBilinearOpModel m({TensorType_UINT8, {2, 2, 2, 1}}); ResizeBilinearOpModel m({TensorType_UINT8, {2, 2, 2, 1}}, {});
m.SetInput<uint8>({ m.SetInput<uint8>({
3, 6, // 3, 6, //
9, 12, // 9, 12, //
@ -336,7 +336,7 @@ TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatchesUInt8) {
} }
TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatchesInt8) { TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatchesInt8) {
ResizeBilinearOpModel m({TensorType_INT8, {2, 2, 2, 1}}); ResizeBilinearOpModel m({TensorType_INT8, {2, 2, 2, 1}}, {});
m.SetInput<int8_t>({ m.SetInput<int8_t>({
3, 6, // 3, 6, //
9, 12, // 9, 12, //
@ -373,7 +373,7 @@ TEST(ResizeBilinearOpTest, TwoDimensionalResizeWithTwoBatchesInt8) {
} }
TEST(ResizeBilinearOpTest, ThreeDimensionalResizeUInt8) { TEST(ResizeBilinearOpTest, ThreeDimensionalResizeUInt8) {
ResizeBilinearOpModel m({TensorType_UINT8, {1, 2, 2, 2}}); ResizeBilinearOpModel m({TensorType_UINT8, {1, 2, 2, 2}}, {});
m.SetInput<uint8>({ m.SetInput<uint8>({
3, 4, 6, 10, // 3, 4, 6, 10, //
10, 12, 14, 16, // 10, 12, 14, 16, //
@ -400,7 +400,7 @@ TEST(ResizeBilinearOpTest, ThreeDimensionalResizeUInt8) {
} }
TEST(ResizeBilinearOpTest, ThreeDimensionalResizeInt8) { TEST(ResizeBilinearOpTest, ThreeDimensionalResizeInt8) {
ResizeBilinearOpModel m({TensorType_INT8, {1, 2, 2, 2}}); ResizeBilinearOpModel m({TensorType_INT8, {1, 2, 2, 2}}, {});
m.SetInput<int8_t>({ m.SetInput<int8_t>({
3, 4, 6, 10, // 3, 4, 6, 10, //
10, 12, 14, 16, // 10, 12, 14, 16, //

30
tensorflow/lite/kernels/resize_nearest_neighbor_test.cc
View File

@ -64,7 +64,7 @@ class ResizeNearestNeighborOpModel : public SingleOpModel {
}; };
TEST(ResizeNearestNeighborOpTest, HorizontalResize) { TEST(ResizeNearestNeighborOpTest, HorizontalResize) {
ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {1, 1, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {1, 1, 2, 1}}, {});
m.SetInput<float>({3, 6}); m.SetInput<float>({3, 6});
m.SetSize({1, 3}); m.SetSize({1, 3});
m.Invoke(); m.Invoke();
@ -80,7 +80,7 @@ TEST(ResizeNearestNeighborOpTest, HorizontalResize) {
} }
TEST(ResizeNearestNeighborOpTest, HorizontalResizeUInt8) { TEST(ResizeNearestNeighborOpTest, HorizontalResizeUInt8) {
ResizeNearestNeighborOpModel m({TensorType_UINT8, {1, 1, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_UINT8, {1, 1, 2, 1}}, {});
m.SetInput<uint8>({3, 6}); m.SetInput<uint8>({3, 6});
m.SetSize({1, 3}); m.SetSize({1, 3});
m.Invoke(); m.Invoke();
@ -96,7 +96,7 @@ TEST(ResizeNearestNeighborOpTest, HorizontalResizeUInt8) {
} }
TEST(ResizeNearestNeighborOpTest, HorizontalResizeInt8) { TEST(ResizeNearestNeighborOpTest, HorizontalResizeInt8) {
ResizeNearestNeighborOpModel m({TensorType_INT8, {1, 1, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_INT8, {1, 1, 2, 1}}, {});
m.SetInput<int8_t>({-3, 6}); m.SetInput<int8_t>({-3, 6});
m.SetSize({1, 3}); m.SetSize({1, 3});
m.Invoke(); m.Invoke();
@ -111,7 +111,7 @@ TEST(ResizeNearestNeighborOpTest, HorizontalResizeInt8) {
} }
TEST(ResizeNearestNeighborOpTest, VerticalResize) { TEST(ResizeNearestNeighborOpTest, VerticalResize) {
ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {1, 2, 1, 1}}); ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {1, 2, 1, 1}}, {});
m.SetInput<float>({3, 9}); m.SetInput<float>({3, 9});
m.SetSize({3, 1}); m.SetSize({3, 1});
m.Invoke(); m.Invoke();
@ -127,7 +127,7 @@ TEST(ResizeNearestNeighborOpTest, VerticalResize) {
} }
TEST(ResizeNearestNeighborOpTest, VerticalResizeUInt8) { TEST(ResizeNearestNeighborOpTest, VerticalResizeUInt8) {
ResizeNearestNeighborOpModel m({TensorType_UINT8, {1, 2, 1, 1}}); ResizeNearestNeighborOpModel m({TensorType_UINT8, {1, 2, 1, 1}}, {});
m.SetInput<uint8>({3, 9}); m.SetInput<uint8>({3, 9});
m.SetSize({3, 1}); m.SetSize({3, 1});
m.Invoke(); m.Invoke();
@ -143,7 +143,7 @@ TEST(ResizeNearestNeighborOpTest, VerticalResizeUInt8) {
} }
TEST(ResizeNearestNeighborOpTest, VerticalResizeInt8) { TEST(ResizeNearestNeighborOpTest, VerticalResizeInt8) {
ResizeNearestNeighborOpModel m({TensorType_INT8, {1, 2, 1, 1}}); ResizeNearestNeighborOpModel m({TensorType_INT8, {1, 2, 1, 1}}, {});
m.SetInput<int8_t>({3, -9}); m.SetInput<int8_t>({3, -9});
m.SetSize({3, 1}); m.SetSize({3, 1});
m.Invoke(); m.Invoke();
@ -158,7 +158,7 @@ TEST(ResizeNearestNeighborOpTest, VerticalResizeInt8) {
} }
TEST(ResizeNearestNeighborOpTest, TwoDimensionalResize) { TEST(ResizeNearestNeighborOpTest, TwoDimensionalResize) {
ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, {});
m.SetInput<float>({ m.SetInput<float>({
3, 6, // 3, 6, //
9, 12 // 9, 12 //
@ -186,7 +186,7 @@ TEST(ResizeNearestNeighborOpTest, TwoDimensionalResize) {
} }
TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeUInt8) { TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeUInt8) {
ResizeNearestNeighborOpModel m({TensorType_UINT8, {1, 2, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_UINT8, {1, 2, 2, 1}}, {});
m.SetInput<uint8>({ m.SetInput<uint8>({
3, 6, // 3, 6, //
9, 12 // 9, 12 //
@ -214,7 +214,7 @@ TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeUInt8) {
} }
TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeInt8) { TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeInt8) {
ResizeNearestNeighborOpModel m({TensorType_INT8, {1, 2, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_INT8, {1, 2, 2, 1}}, {});
m.SetInput<int8_t>({ m.SetInput<int8_t>({
3, -6, // 3, -6, //
9, 12 // 9, 12 //
@ -241,7 +241,7 @@ TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeInt8) {
} }
TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatches) { TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatches) {
ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {2, 2, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {2, 2, 2, 1}}, {});
m.SetInput<float>({ m.SetInput<float>({
3, 6, // 3, 6, //
9, 12, // 9, 12, //
@ -279,7 +279,7 @@ TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatches) {
} }
TEST(ResizeNearestNeighborOpTest, ThreeDimensionalResize) { TEST(ResizeNearestNeighborOpTest, ThreeDimensionalResize) {
ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {1, 2, 2, 2}}); ResizeNearestNeighborOpModel m({TensorType_FLOAT32, {1, 2, 2, 2}}, {});
m.SetInput<float>({ m.SetInput<float>({
3, 4, 6, 10, // 3, 4, 6, 10, //
9, 10, 12, 16, // 9, 10, 12, 16, //
@ -307,7 +307,7 @@ TEST(ResizeNearestNeighborOpTest, ThreeDimensionalResize) {
} }
TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatchesUInt8) { TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatchesUInt8) {
ResizeNearestNeighborOpModel m({TensorType_UINT8, {2, 2, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_UINT8, {2, 2, 2, 1}}, {});
m.SetInput<uint8>({ m.SetInput<uint8>({
3, 6, // 3, 6, //
9, 12, // 9, 12, //
@ -345,7 +345,7 @@ TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatchesUInt8) {
} }
TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatchesInt8) { TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatchesInt8) {
ResizeNearestNeighborOpModel m({TensorType_INT8, {2, 2, 2, 1}}); ResizeNearestNeighborOpModel m({TensorType_INT8, {2, 2, 2, 1}}, {});
m.SetInput<int8_t>({ m.SetInput<int8_t>({
3, 6, // 3, 6, //
9, -12, // 9, -12, //
@ -382,7 +382,7 @@ TEST(ResizeNearestNeighborOpTest, TwoDimensionalResizeWithTwoBatchesInt8) {
} }
TEST(ResizeNearestNeighborOpTest, ThreeDimensionalResizeUInt8) { TEST(ResizeNearestNeighborOpTest, ThreeDimensionalResizeUInt8) {
ResizeNearestNeighborOpModel m({TensorType_UINT8, {1, 2, 2, 2}}); ResizeNearestNeighborOpModel m({TensorType_UINT8, {1, 2, 2, 2}}, {});
m.SetInput<uint8>({ m.SetInput<uint8>({
3, 4, 6, 10, // 3, 4, 6, 10, //
10, 12, 14, 16, // 10, 12, 14, 16, //
@ -410,7 +410,7 @@ TEST(ResizeNearestNeighborOpTest, ThreeDimensionalResizeUInt8) {
} }
TEST(ResizeNearestNeighborOpTest, ThreeDimensionalResizeInt8) { TEST(ResizeNearestNeighborOpTest, ThreeDimensionalResizeInt8) {
ResizeNearestNeighborOpModel m({TensorType_INT8, {1, 2, 2, 2}}); ResizeNearestNeighborOpModel m({TensorType_INT8, {1, 2, 2, 2}}, {});
m.SetInput<int8_t>({ m.SetInput<int8_t>({
3, 4, -6, 10, // 3, 4, -6, 10, //
10, 12, -14, 16, // 10, 12, -14, 16, //