| /external/tensorflow/tensorflow/lite/kernels/ |
| strided_slice_test.cc | 24 using ::testing::ElementsAreArray;
97 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2}));
98 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3}));
108 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({0}));
118 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2}));
119 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3}));
129 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3}));
130 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3}));
140 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1}));
141 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2}))
[all...] |
| one_hot_test.cc | 27 using ::testing::ElementsAreArray;
71 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3}));
73 ElementsAreArray({1.f, 0.f, 0.f, 0.f, 1.f, 0.f, 0.f, 0.f, 1.f}));
82 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3}));
83 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 0, 0, 0, 1, 0, 0, 0, 1}));
92 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 3}));
94 ElementsAreArray({true, false, false, false, true, false, false,
104 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({3, 1}));
105 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 0, 0}));
114 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 4}))
[all...] |
| space_to_batch_nd_test.cc | 25 using ::testing::ElementsAreArray;
125 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 2, 1}));
126 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7,
137 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 2, 1}));
138 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7,
147 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8, 1, 2, 1}));
148 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7,
159 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8, 1, 2, 1}));
160 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7,
169 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1}))
[all...] |
| zeros_like_test.cc | 24 using ::testing::ElementsAreArray;
49 ElementsAreArray({0.0, 0.0, 0.0, 0.0, 0.0, 0.0}));
50 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({2, 3}));
58 ElementsAreArray({0, 0, 0, 0}));
59 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 2, 2, 1}));
67 ElementsAreArray({0, 0, 0, 0}));
68 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 2, 2, 1}));
|
| expand_dims_test.cc | 26 using ::testing::ElementsAreArray;
58 EXPECT_THAT(m.GetValuesFloat(), ElementsAreArray(values));
59 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 2}));
63 EXPECT_THAT(m.GetValuesFloat(), ElementsAreArray(values));
64 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 2}));
68 EXPECT_THAT(m.GetValuesFloat(), ElementsAreArray(values));
69 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 1}));
73 EXPECT_THAT(m.GetValuesFloat(), ElementsAreArray(values));
74 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 1}));
|
| select_test.cc | 24 using ::testing::ElementsAreArray;
69 ElementsAreArray({false, true, false, true}));
70 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4}));
82 EXPECT_THAT(model.GetOutput<float>(), ElementsAreArray({0.1, 0.6, 0.3, 0.8}));
83 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4}));
95 EXPECT_THAT(model.GetOutput<uint8_t>(), ElementsAreArray({5, 2, 7, 8}));
96 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4}));
108 EXPECT_THAT(model.GetOutput<int8_t>(), ElementsAreArray({5, -2, 7, -8}));
109 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1, 4}));
121 EXPECT_THAT(model.GetOutput<int16_t>(), ElementsAreArray({5, 2, 7, 8}))
[all...] |
| topk_v2_test.cc | 26 using ::testing::ElementsAreArray;
97 EXPECT_THAT(m.GetIndexes(), ElementsAreArray({1, 0, 0, 1}));
99 ElementsAreArray(ArrayFloatNear({0.2, -2.0, 0.8, 0.1})));
107 EXPECT_THAT(m.GetIndexes(), ElementsAreArray({2, 0, 0, 1}));
109 ElementsAreArray(ArrayFloatNear({0.2, -2.0, 0.8, 0.1})));
116 EXPECT_THAT(m.GetIndexes(), ElementsAreArray({3, 0, 0, 1}));
118 ElementsAreArray(ArrayFloatNear({0.2, -2.0, 0.8, 0.1})));
126 EXPECT_THAT(m.GetIndexes(), ElementsAreArray({4, 3}));
127 EXPECT_THAT(m.GetValuesFloat(), ElementsAreArray(ArrayFloatNear({0.8, 0.2})));
135 EXPECT_THAT(m.GetIndexes(), ElementsAreArray({2, 1, 0, 1}))
[all...] |
| gather_test.cc | 25 using ::testing::ElementsAreArray;
75 ElementsAreArray(ArrayFloatNear({0.7, 0.8, -2, 0.2})));
84 ElementsAreArray(ArrayFloatNear({0.7, 0.8})));
85 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2}));
95 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({2.0})));
104 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({3.0})));
105 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1}));
114 ElementsAreArray(ArrayFloatNear({2.0, 1.0})));
115 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2}));
125 ElementsAreArray(ArrayFloatNear({-2, 0.2, 0.7, 0.8, -2, 0.2, 0.7, 0.8})))
[all...] |
| reduce_test.cc | 24 using ::testing::ElementsAreArray;
243 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2}));
244 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({12, 13})));
255 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3, 1}));
257 ElementsAreArray(ArrayFloatNear({10.5, 12.5, 14.5})));
270 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 1, 2}));
272 ElementsAreArray(ArrayFloatNear({6, 7, 18, 19})));
283 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 3}));
285 ElementsAreArray(ArrayFloatNear({0.25098, 0.25098, 0.25098},
299 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 1, 1, 9}))
[all...] |
| elementwise_test.cc | 25 using ::testing::ElementsAreArray;
64 ElementsAreArray(ArrayFloatNear({0, 0, 0, 0.84147})));
65 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1}));
73 ElementsAreArray(ArrayFloatNear({1, -1, -1, 0.54030})));
74 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1}));
82 ElementsAreArray(ArrayFloatNear({0, 1.14473, 0, 0})));
83 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1}));
93 EXPECT_THAT(m.ExtractVector<float>(m.output()), ElementsAreArray({
104 ElementsAreArray(ArrayFloatNear({0, 1, 1.41421, 2})));
105 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1}))
[all...] |
| test_util_test.cc | 21 using ::testing::ElementsAreArray;
27 EXPECT_THAT(q_data, ElementsAreArray(expected));
34 EXPECT_THAT(q_data, ElementsAreArray(expected));
41 EXPECT_THAT(q_data, ElementsAreArray(expected));
|
| tile_test.cc | 25 using ::testing::ElementsAreArray;
65 ElementsAreArray({1.f, 2.f, 3.f, 1.f, 2.f, 3.f}));
80 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({
94 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3}));
111 ElementsAreArray({11.f, 12.f, 13.f, 21.f, 22.f, 23.f, 11.f, 12.f, 13.f,
115 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 6, 3}));
130 EXPECT_THAT(m.GetOutput<uint8_t>(), ElementsAreArray({
144 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3}));
159 EXPECT_THAT(m.GetOutput<int32_t>(), ElementsAreArray({
173 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 3}))
[all...] |
| arg_min_max_test.cc | 24 using ::testing::ElementsAreArray;
85 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1}));
86 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1}));
96 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1}));
97 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1}));
107 EXPECT_THAT(model.GetOutput(), ElementsAreArray({2}));
108 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1}));
118 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1}));
119 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({1, 1, 1}));
129 EXPECT_THAT(model.GetOutput(), ElementsAreArray({3, 1}))
[all...] |
| l2norm_test.cc | 24 using ::testing::ElementsAreArray;
77 ElementsAreArray({-0.55, 0.3, 0.35, 0.6, -0.35, 0.05}));
90 ElementsAreArray({
103 ElementsAreArray({58, 166, 173, 205, 83, 134}));
105 ElementsAreArray(
115 ElementsAreArray({-70, 38, 45, 77, -45, 6}));
118 ElementsAreArray(
133 ElementsAreArray({
139 ElementsAreArray(ArrayFloatNear(
158 EXPECT_THAT(m.GetOutput<int8_t>(), ElementsAreArray({
[all...] |
| unique_test.cc | 24 using ::testing::ElementsAreArray;
57 EXPECT_THAT(model.GetOutput(), ElementsAreArray({5}));
58 EXPECT_THAT(model.GetIndexesOutput(), ElementsAreArray({0}));
67 EXPECT_THAT(model.GetOutput(), ElementsAreArray({5, 2, 3, 51, 6, 72, 7, 8}));
69 ElementsAreArray({0, 1, 2, 3, 4, 5, 6, 7}));
77 EXPECT_THAT(model.GetOutput(), ElementsAreArray({5}));
79 ElementsAreArray({0, 0, 0, 0, 0, 0, 0}));
87 EXPECT_THAT(model.GetOutput(), ElementsAreArray({2, 3, 5, 7}));
89 ElementsAreArray({0, 1, 2, 3, 0, 3, 1}));
97 EXPECT_THAT(model.GetOutput(), ElementsAreArray({2, 3, 5, 7}))
[all...] |
| pad_test.cc | 25 using ::testing::ElementsAreArray;
217 ElementsAreArray({0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 0,
219 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 2, 4, 1}));
229 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, 0, 0, 1, 2, 0, 0, 3, 4,
231 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1}));
240 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, 0, 0, 0, 0, 0}));
241 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 4, 1}));
250 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, 0, 0, 0, 0, 0}));
251 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 2, 1}));
259 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2, 3, 0, 0}))
[all...] |
| shape_test.cc | 27 using ::testing::ElementsAreArray;
59 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 3, 1, 3, 5}));
60 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({5}));
68 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 3, 1, 3, 5}));
69 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({5}));
77 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({0}));
84 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 0}));
85 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2}));
|
| squeeze_test.cc | 24 using ::testing::ElementsAreArray;
67 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({24}));
70 ElementsAreArray({1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0,
83 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 24}));
86 ElementsAreArray({1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0,
99 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({24}));
102 ElementsAreArray({1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0,
114 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3.85}));
|
| resize_bilinear_test.cc | 24 using ::testing::ElementsAreArray;
72 ElementsAreArray(ArrayFloatNear({3, 5, 6})));
78 ElementsAreArray(ArrayFloatNear({3, 5, 6})));
87 ElementsAreArray(ArrayFloatNear({3, 5, 6})));
93 ElementsAreArray(ArrayFloatNear({3, 5, 6})));
102 ElementsAreArray(ArrayFloatNear({3, 5, 6})));
108 ElementsAreArray(ArrayFloatNear({3, 5, 6})));
117 ElementsAreArray(ArrayFloatNear({3, 7, 9})));
123 ElementsAreArray(ArrayFloatNear({3, 7, 9})));
132 ElementsAreArray(ArrayFloatNear({3, 7, 9})))
[all...] |
| resize_nearest_neighbor_test.cc | 24 using ::testing::ElementsAreArray;
72 ElementsAreArray(ArrayFloatNear({3, 3, 6})));
79 ElementsAreArray(ArrayFloatNear({3, 3, 6})));
88 ElementsAreArray(ArrayFloatNear({3, 3, 6})));
95 ElementsAreArray(ArrayFloatNear({3, 3, 6})));
104 ElementsAreArray(ArrayFloatNear({-3, -3, 6})));
110 ElementsAreArray(ArrayFloatNear({-3, -3, 6})));
119 ElementsAreArray(ArrayFloatNear({3, 3, 9})));
126 ElementsAreArray(ArrayFloatNear({3, 3, 9})));
135 ElementsAreArray(ArrayFloatNear({3, 3, 9})))
[all...] |
| slice_test.cc | 24 using ::testing::ElementsAreArray;
71 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2}));
72 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3}));
82 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2}));
83 EXPECT_THAT(m.GetOutput(), ElementsAreArray({4, 5}));
93 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3, 2}));
95 ElementsAreArray({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}));
105 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 1, 1}));
106 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3, 4}));
116 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 1, 1}))
[all...] |
| activations_test.cc | 25 using ::testing::ElementsAreArray;
129 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({
143 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
157 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
171 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
185 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({
204 ElementsAreArray(ArrayFloatNear(
211 ElementsAreArray({128, 128, 160, 192, 176, 128, 224, 144}));
226 EXPECT_THAT(m.GetDequantizedOutput<int8_t>(), ElementsAreArray(ArrayFloatNear(
233 ElementsAreArray({0, 0, 32, 64, 48, 0, 96, 16}))
[all...] |
| sparse_to_dense_test.cc | 26 using ::testing::ElementsAreArray;
73 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, 7, 0}));
74 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({5}));
85 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2, 0, 4, 0, 6, 0}));
86 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({7}));
98 ElementsAreArray({2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
100 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3}));
113 ElementsAreArray({2, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
115 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 3, 3}));
128 ElementsAreArray({2, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1
[all...] |
| transpose_conv_test.cc | 36 using ::testing::ElementsAreArray;
109 ElementsAreArray({29, 62, 83, 75, 99, 192, 237, 198, 207, 372,
112 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1}));
138 ElementsAreArray({184, 412, 568, 528, 678, 1347, 1689, 1434, 1494,
140 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1}));
167 ElementsAreArray({5, 22, 59, 101, 114, 83, 52, 184, 412,
171 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 6, 6, 1}));
195 ElementsAreArray({1, 2, 5, 4, 6, 4, 5, 14, 10, 12, 10, 14, 36,
197 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 5, 5, 1}));
221 ElementsAreArray({1, 2, 3, 4, 7, 10, 6, 8, 10, 12, 7, 8, 9
[all...] |
| /external/tensorflow/tensorflow/lite/models/smartreply/ops/ |
| predict_test.cc | 33 using ::testing::ElementsAreArray;
115 EXPECT_THAT(m.GetLabel(), ElementsAreArray({12, 11}));
116 EXPECT_THAT(m.GetWeight(), ElementsAreArray(ArrayFloatNear({0.1, 0.05})));
126 EXPECT_THAT(m.GetLabel(), ElementsAreArray({12}));
127 EXPECT_THAT(m.GetWeight(), ElementsAreArray(ArrayFloatNear({0.1})));
137 EXPECT_THAT(m.GetLabel(), ElementsAreArray({12, -1}));
138 EXPECT_THAT(m.GetWeight(), ElementsAreArray(ArrayFloatNear({0.1, 0})));
148 EXPECT_THAT(m.GetLabel(), ElementsAreArray({-1, -1}));
149 EXPECT_THAT(m.GetWeight(), ElementsAreArray(ArrayFloatNear({0, 0})));
159 EXPECT_THAT(m.GetLabel(), ElementsAreArray({11, -1}))
[all...] |
