| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 65 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1c38552f0111::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::__anon908f57500111::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::__anonee8c12e90111::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::__anon75467ea60111::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::__anoncae41a510111::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::__anon667d06810111::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_test.cc | 44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon0a22c2ad0111::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 | reverse_sequence_test.cc | 46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anond053f1e50111::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 | arg_min_max_test.cc | 40 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon0dc779d30111::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::__anon580bf43d0111::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::__anon92c241bf0111::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::__anone8b6eaff0111::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::__anon2ee682910111::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::__anon60c92a720111::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::__anona7b28c900111::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::__anon5bb25c730111::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::__anona7a2c89e0111::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::__anon6134f40c0111::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::__anona42cd54f0111::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::__anon558e5be50111::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::__anon5316d1870111::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::__anon1eea4e7d0111::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::__anon459c72dd0111::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::__anonac5345a20111::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 | fake_quant_test.cc | 46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonbbb380900111::FakeQuantOpModel
60 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); in TEST()
73 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); in TEST()
86 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); in TEST()
99 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); in TEST()
|