| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 65 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anone1578cc60111::StridedSliceOpModel
97 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
108 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({0})); in TEST()
118 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
129 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
140 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
151 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
162 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
173 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
184 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
[all …]
|
| D | pack_test.cc | 48 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anone90cb1070111::PackOpModel
61 EXPECT_THAT(model.GetOutputShape(), ElementsAre(3, 2)); in TEST()
71 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 3)); in TEST()
80 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 2, 3)); in TEST()
91 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 2, 2, 2, 2)); in TEST()
105 EXPECT_THAT(model.GetOutputShape(), ElementsAre(3, 2)); in TEST()
115 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 3)); in TEST()
124 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 2, 3)); in TEST()
136 EXPECT_THAT(model.GetOutputShape(), ElementsAre(3, 2)); in TEST()
147 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 3)); in TEST()
[all …]
|
| D | reduce_test.cc | 47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5bc6b6600111::BaseOpModel
243 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
255 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3, 1})); in TEST()
270 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 1, 2})); in TEST()
283 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 3})); in TEST()
299 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 9})); in TEST()
314 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 2})); in TEST()
326 EXPECT_THAT(m.GetOutputShape(), IsEmpty()); in TEST()
341 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
356 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3, 1})); in TEST()
[all …]
|
| D | pad_test.cc | 52 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonf413fd7d0111::PadOpModel
219 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 2, 4, 1})); in TEST()
231 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
241 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 4, 1})); in TEST()
251 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 2, 1})); in TEST()
260 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({5})); in TEST()
271 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
284 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 4, 6, 1})); in TEST()
295 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 7, 1})); in TEST()
307 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 7, 1})); in TEST()
[all …]
|
| D | space_to_batch_nd_test.cc | 48 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonf57345680111::SpaceToBatchNDOpModel
125 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 2, 1})); in TEST()
137 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 2, 1})); in TEST()
147 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8, 1, 2, 1})); in TEST()
159 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8, 1, 2, 1})); in TEST()
169 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1})); in TEST()
183 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1})); in TEST()
195 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 4, 1})); in TEST()
210 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 4, 1})); in TEST()
245 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1})); in TEST_F()
[all …]
|
| D | slice_test.cc | 55 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anoncea597180111::SliceOpModel
71 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
82 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2})); in TEST()
93 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3, 2})); in TEST()
105 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 1, 1})); in TEST()
116 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 1, 1})); in TEST()
129 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 3, 1})); in TEST()
140 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 3, 1})); in TEST()
151 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 3, 1})); in TEST()
162 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 3, 1})); in TEST()
[all …]
|
| D | reverse_sequence_test.cc | 46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon938e961c0111::ReverseSequenceOpModel
63 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
78 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
93 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
108 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
123 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
138 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
153 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
168 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
183 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
[all …]
|
| D | reverse_test.cc | 44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1eb1d5040111::ReverseOpModel
60 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
73 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
88 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
101 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
116 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
129 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
144 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
157 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
172 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
[all …]
|
| D | arg_min_max_test.cc | 40 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1db5dfaa0111::ArgBaseOpModel
86 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST()
97 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST()
108 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST()
119 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST()
130 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 2})); in TEST()
141 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 4})); in TEST()
152 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 2})); in TEST()
163 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST()
174 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST()
[all …]
|
| D | one_hot_test.cc | 58 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon6c9b06940111::OneHotOpModel
71 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
82 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
92 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
104 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 1})); in TEST()
114 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 4})); in TEST()
127 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({4, 3})); in TEST()
141 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 4})); in TEST()
155 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 2, 3})); in TEST()
171 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
|
| D | comparisons_test.cc | 51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anona2b0a7960111::ComparisonOpModel
103 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
114 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
125 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
136 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
148 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 2, 4)); in TEST()
159 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
170 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
181 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
192 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
[all …]
|
| D | tile_test.cc | 51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon413444b60111::TileOpModel
94 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3})); in TEST()
115 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 6, 3})); in TEST()
144 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3})); in TEST()
173 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3})); in TEST()
186 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3})); in TEST()
215 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3})); in TEST()
244 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3})); in TEST()
|
| D | select_test.cc | 50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon9c2126080111::SelectOpModel
70 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
83 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
96 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
109 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
122 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
135 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
147 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 1, 2, 1})); in TEST()
159 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 2, 2, 1})); in TEST()
|
| D | floor_mod_test.cc | 43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anond5662e890111::FloorModModel
58 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
69 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
79 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
89 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
101 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
112 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
123 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
|
| D | transpose_test.cc | 162 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1c4f90a70111::TransposeOpModel
227 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
236 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
244 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); in TEST()
253 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); in TEST()
262 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 3})); in TEST()
274 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 3})); in TEST()
291 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 3, 1})); in TEST()
300 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 3, 1})); in TEST()
308 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 2, 1, 1})); in TEST()
[all …]
|
| D | space_to_depth_test.cc | 46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonc4b886ea0111::SpaceToDepthOpModel
66 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 1, 1, 8)); in TEST()
74 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
82 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
91 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 1, 1, 12)); in TEST()
101 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 2, 2, 4)); in TEST()
|
| D | range_test.cc | 44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon0fcb59350111::RangeOpModel
59 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
69 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
79 EXPECT_THAT(model.GetOutputShape(), ElementsAre(3)); in TEST()
89 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
99 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
109 EXPECT_THAT(model.GetOutputShape(), ElementsAre(3)); in TEST()
|
| D | gather_test.cc | 61 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonce6f97830111::GatherOpModel
85 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
96 EXPECT_TRUE(m.GetOutputShape().empty()); in TEST()
105 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
115 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2})); in TEST()
146 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 3})); in TEST()
157 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2})); in TEST()
169 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 2})); in TEST()
181 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 2})); in TEST()
192 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2})); in TEST()
[all …]
|
| D | sparse_to_dense_test.cc | 55 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon2df84ea60111::SparseToDenseOpModel
74 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({5})); in TEST()
86 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({7})); in TEST()
100 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
115 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
130 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
145 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
|
| D | transpose_conv_test.cc | 71 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonbe94865c0111::TransposeConvOpModel
112 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST_P()
140 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST_P()
171 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 6, 6, 1})); in TEST_P()
197 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 5, 5, 1})); in TEST_P()
225 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 5, 5, 2})); in TEST_P()
252 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3, 4, 1})); in TEST_P()
|
| D | pow_test.cc | 44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anond1e4505e0111::PowOpModel
59 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
70 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
81 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
94 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
106 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
|
| D | rank_test.cc | 45 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon7767a8340111::RankOpModel
57 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
65 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
73 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
81 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
|
| D | expand_dims_test.cc | 44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon558ad8b40111::ExpandDimsOpModel
59 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 2})); in TEST()
64 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 2})); in TEST()
69 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 1})); in TEST()
74 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 1})); in TEST()
|
| D | shape_test.cc | 47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon147bd6390111::ShapeOpModel
60 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({5})); in TEST()
69 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({5})); in TEST()
77 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({0})); in TEST()
85 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2})); in TEST()
|
| D | squeeze_test.cc | 56 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon9031b06a0111::FloatSqueezeOpModel
67 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({24})); in TEST()
83 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 24})); in TEST()
99 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({24})); in TEST()
113 EXPECT_THAT(m.GetOutputShape(), IsEmpty()); in TEST()
|