| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 77 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon53cd168d0111::StridedSliceOpModel
116 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({0})); in TYPED_TEST()
126 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TYPED_TEST()
141 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({32768})); in TYPED_TEST()
152 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({0})); in TYPED_TEST()
162 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TYPED_TEST()
173 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TYPED_TEST()
184 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TYPED_TEST()
195 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TYPED_TEST()
206 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TYPED_TEST()
[all …]
|
| D | slice_test.cc | 76 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonf6cd49df0111::SliceOpModel
92 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST_P()
102 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2})); in TEST_P()
112 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3, 2})); in TEST_P()
123 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 1, 1, 1})); in TEST_P()
133 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 1, 1})); in TEST_P()
143 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 1, 1})); in TEST_P()
155 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 3, 1})); in TEST_P()
165 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 3, 1})); in TEST_P()
175 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 3, 1})); in TEST_P()
[all …]
|
| D | broadcast_to_test.cc | 68 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonae4de6640111::BroadcastToOpModel
121 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4})); in TYPED_TEST()
129 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 2, 2})); in TYPED_TEST()
138 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 1, 1, 1, 2, 2})); in TYPED_TEST()
147 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4})); in TYPED_TEST()
156 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 2, 2})); in TYPED_TEST()
165 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 1, 1, 1, 2, 2})); in TYPED_TEST()
173 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 2, 2})); in TYPED_TEST()
185 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 2, 2})); in TYPED_TEST()
196 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 1, 3, 2, 2})); in TYPED_TEST()
[all …]
|
| D | pack_test.cc | 53 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon3015a6ee0111::PackOpModel
66 EXPECT_THAT(model.GetOutputShape(), ElementsAre(3, 2)); in TEST()
76 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 3)); in TEST()
86 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 3)); in TEST()
95 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 2, 3)); in TEST()
106 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 2, 2, 2, 2)); in TEST()
120 EXPECT_THAT(model.GetOutputShape(), ElementsAre(3, 2)); in TEST()
130 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 3)); in TEST()
140 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 3)); in TEST()
149 EXPECT_THAT(model.GetOutputShape(), ElementsAre(2, 2, 3)); in TEST()
[all …]
|
| D | space_to_batch_nd_test.cc | 52 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon57965aaf0111::SpaceToBatchNDOpModel
132 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 2, 1})); in TEST()
144 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 2, 1})); in TEST()
154 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8, 1, 2, 1})); in TEST()
166 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8, 1, 2, 1})); in TEST()
176 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1})); in TEST()
190 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1})); in TEST()
202 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 4, 1})); in TEST()
217 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 4, 1})); in TEST()
252 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1})); in TEST_F()
[all …]
|
| D | transpose_test.cc | 221 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anona3114eee0111::TransposeOpModel
286 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
295 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
303 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); in TEST()
312 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 4})); in TEST()
323 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 6})); in TEST()
334 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); in TEST()
343 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 3})); in TEST()
355 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 3})); in TEST()
365 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
[all …]
|
| D | reduce_test.cc | 51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon88e4c2070111::BaseOpModel
272 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
284 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3, 1})); in TEST()
296 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 0, 1})); in TEST()
309 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 1, 2})); in TEST()
322 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 3})); in TEST()
340 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 18})); in TEST()
356 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 2})); in TEST()
368 EXPECT_THAT(m.GetOutputShape(), IsEmpty()); in TEST()
378 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
[all …]
|
| D | select_test.cc | 50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonc93f31af0111::BaseSelectOpModel
97 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
110 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
123 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
136 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
149 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
162 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4})); in TEST()
174 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 1, 2, 1})); in TEST()
189 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 1, 2, 1})); in TEST()
204 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 1, 2, 1})); in TEST()
[all …]
|
| D | one_hot_test.cc | 61 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon3d7a871b0111::OneHotOpModel
74 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
85 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
95 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
105 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
115 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3})); in TEST()
127 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 1})); in TEST()
137 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 4})); in TEST()
150 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({4, 3})); in TEST()
164 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 4})); in TEST()
[all …]
|
| D | scatter_nd_test.cc | 64 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonb54fb7560111::ScatterNdOpModel
81 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8})); in TEST()
95 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 4, 4})); in TEST()
115 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({10, 4})); in TEST()
140 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 2, 2})); in TEST()
158 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3, 5})); in TEST()
201 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3, 5})); in TEST()
221 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3, 5})); in TEST()
241 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3, 5})); in TEST()
261 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3, 5})); in TEST()
[all …]
|
| D | reverse_test.cc | 47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonef91558b0111::ReverseOpModel
63 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
76 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
91 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
104 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
119 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
132 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
147 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
160 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
174 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4)); in TEST()
[all …]
|
| D | pad_test.cc | 56 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon535252840111::PadOpModel
223 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 2, 4, 1})); in TEST()
235 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
245 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 4, 1})); in TEST()
255 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 2, 1})); in TEST()
264 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({5})); in TEST()
275 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
298 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 4, 6, 1})); in TEST()
309 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 7, 1})); in TEST()
321 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 7, 1})); in TEST()
[all …]
|
| D | batch_matmul_test.cc | 59 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function in tflite::__anon83a19b600111::BatchMatMulOpModel
89 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 2, 4})); in TEST_P()
101 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 2, 4})); in TEST_P()
113 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 2, 4})); in TEST_P()
129 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 2, 4})); in TEST_P()
145 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 2, 4})); in TEST_P()
161 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 2, 4})); in TEST_P()
185 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 3, 3, 4})); in TEST_P()
209 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 3, 3, 4})); in TEST_P()
232 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 3, 3, 4})); in TEST_P()
[all …]
|
| D | reverse_sequence_test.cc | 49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon39709e830111::ReverseSequenceOpModel
66 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
81 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
96 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
111 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
126 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
141 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
156 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
171 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
186 EXPECT_THAT(model.GetOutputShape(), ElementsAre(4, 3, 2)); in TEST()
[all …]
|
| D | batch_to_space_nd_test.cc | 53 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon01852b6f0111::BatchToSpaceNDOpModel
116 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
127 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
137 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 2, 1})); in TEST()
151 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 0, 1})); in TEST()
161 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
173 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
198 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 0, 4, 1})); in TEST()
218 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 8, 1})); in TEST()
228 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 6, 1})); in TEST()
[all …]
|
| D | arg_min_max_test.cc | 61 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5be04bb10111::ArgBaseOpModel
149 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST_P()
159 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST_P()
169 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST_P()
179 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST_P()
189 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 2})); in TEST_P()
199 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 4})); in TEST_P()
209 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 2})); in TEST_P()
219 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST_P()
229 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1})); in TEST_P()
[all …]
|
| D | fill_test.cc | 68 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon0a5536560111::FillOpModel
103 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon0a5536560111::QuantizedFillOpModel
118 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); in TEST_P()
128 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 4})); in TEST_P()
137 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 2})); in TEST_P()
146 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 2})); in TEST_P()
153 EXPECT_THAT(m.GetOutputShape(), IsEmpty()); in TEST_P()
162 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 2})); in TEST_P()
171 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 2})); in TEST()
179 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 2})); in TEST_P()
[all …]
|
| D | sparse_to_dense_test.cc | 56 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon13f7662d0111::SparseToDenseOpModel
75 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({5})); in TEST()
87 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({7})); in TEST()
101 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
116 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
131 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
146 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
161 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
176 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
190 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3})); in TEST()
|
| D | comparisons_test.cc | 58 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anone0db139d0111::ComparisonOpModel
110 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
121 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
132 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
146 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4, 1)); in TEST()
157 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
169 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 2, 4)); in TEST()
183 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
194 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
205 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
[all …]
|
| D | floor_mod_test.cc | 46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5c27ecd00111::FloorModModel
61 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
72 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
82 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
92 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
104 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
115 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
126 EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); in TEST()
|
| D | rank_test.cc | 49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonbe709e1b0111::RankOpModel
65 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
71 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
78 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
85 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
92 EXPECT_TRUE(model.GetOutputShape().empty()); in TEST()
|
| D | shape_test.cc | 50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon3ca389000111::ShapeOpModel
67 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({5})); in TEST()
73 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({5})); in TEST()
81 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({5})); in TEST()
88 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({0})); in TEST()
95 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2})); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | slice_test.cc | 48 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::SliceOpModel
64 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
77 auto reference_output_shape = m.GetOutputShape(); in TEST()
79 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray(reference_output_shape)); in TEST()
92 auto reference_output_shape = m.GetOutputShape(); in TEST()
94 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray(reference_output_shape)); in TEST()
108 auto reference_output_shape = m.GetOutputShape(); in TEST()
110 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray(reference_output_shape)); in TEST()
124 auto reference_output_shape = m.GetOutputShape(); in TEST()
126 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray(reference_output_shape)); in TEST()
[all …]
|
| D | strided_slice_test.cc | 54 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::StridedSliceOpModel
77 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
94 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
111 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
128 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3})); in TEST()
145 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3})); in TEST()
162 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3})); in TEST()
179 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4})); in TEST()
196 EXPECT_TRUE(m.GetOutputShape().empty()); in TEST()
213 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3, 2})); in TEST()
[all …]
|
| D | pack_test.cc | 36 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::PackOpModel
72 auto ref_output_shape = model.GetOutputShape(); in TYPED_TEST()
76 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray(ref_output_shape)); in TYPED_TEST()
87 auto ref_output_shape = model.GetOutputShape(); in TYPED_TEST()
91 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray(ref_output_shape)); in TYPED_TEST()
102 auto ref_output_shape = model.GetOutputShape(); in TYPED_TEST()
106 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray(ref_output_shape)); in TYPED_TEST()
116 auto ref_output_shape = model.GetOutputShape(); in TYPED_TEST()
120 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray(ref_output_shape)); in TYPED_TEST()
|