| /external/tensorflow/tensorflow/lite/kernels/ |
| where_test.cc | 49 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); }
56 EXPECT_THAT(m.GetOutput().size(), 0);
63 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2}));
72 EXPECT_EQ(m.GetOutput().size(), 0);
79 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, //
89 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, //
102 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, //
116 EXPECT_EQ(m.GetOutput().size(), 0);
124 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, //
134 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, /
[all...] |
| resize_bilinear_test.cc | 56 std::vector<T> GetOutput() {
71 EXPECT_THAT(m.GetOutput<float>(),
77 EXPECT_THAT(const_m.GetOutput<float>(),
86 EXPECT_THAT(m.GetOutput<uint8>(),
92 EXPECT_THAT(const_m.GetOutput<uint8>(),
101 EXPECT_THAT(m.GetOutput<int8_t>(),
107 EXPECT_THAT(const_m.GetOutput<int8_t>(),
116 EXPECT_THAT(m.GetOutput<float>(),
122 EXPECT_THAT(const_m.GetOutput<float>(),
131 EXPECT_THAT(m.GetOutput<uint8>()
[all...] |
| resize_nearest_neighbor_test.cc | 56 std::vector<T> GetOutput() {
71 EXPECT_THAT(m.GetOutput<float>(),
78 EXPECT_THAT(const_m.GetOutput<float>(),
87 EXPECT_THAT(m.GetOutput<uint8>(),
94 EXPECT_THAT(const_m.GetOutput<uint8>(),
103 EXPECT_THAT(m.GetOutput<int8_t>(),
109 EXPECT_THAT(const_m.GetOutput<int8_t>(),
118 EXPECT_THAT(m.GetOutput<float>(),
125 EXPECT_THAT(const_m.GetOutput<float>(),
134 EXPECT_THAT(m.GetOutput<uint8>()
[all...] |
| skip_gram_test.cc | 48 std::vector<string> GetOutput() {
70 EXPECT_THAT(m.GetOutput(), testing::UnorderedElementsAreArray(
79 EXPECT_THAT(m.GetOutput(),
89 EXPECT_THAT(m.GetOutput(),
103 EXPECT_THAT(m.GetOutput(),
122 m.GetOutput(),
133 EXPECT_THAT(m.GetOutput(),
147 EXPECT_THAT(m.GetOutput(),
163 EXPECT_THAT(m.GetOutput(),
187 m.GetOutput(),
[all...] |
| local_response_norm_test.cc | 44 std::vector<float> GetOutput() { return ExtractVector<float>(output_); }
58 m.GetOutput(),
68 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear(
79 m.GetOutput(),
89 m.GetOutput(),
|
| rank_test.cc | 44 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); }
56 EXPECT_THAT(model.GetOutput(), ElementsAreArray({5}));
64 EXPECT_THAT(model.GetOutput(), ElementsAreArray({5}));
72 EXPECT_THAT(model.GetOutput(), ElementsAreArray({0}));
80 EXPECT_THAT(model.GetOutput(), ElementsAreArray({2}));
|
| gather_nd_test.cc | 50 std::vector<T> GetOutput() {
68 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 2.2}));
77 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({2.1, 2.2, 1.1, 1.2}));
87 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({2.1, 2.2, 1.1, 1.2}));
97 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 2.2}));
106 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 1.1}));
118 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({-1.2, -4.1}));
130 EXPECT_THAT(m.GetOutput<float>(),
144 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({-1.2, -4.1}));
156 EXPECT_THAT(m.GetOutput<float>()
[all...] |
| gather_test.cc | 54 std::vector<T> GetOutput() {
74 EXPECT_THAT(m.GetOutput<float>(),
83 EXPECT_THAT(m.GetOutput<float>(),
95 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({2.0})));
104 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({3.0})));
113 EXPECT_THAT(m.GetOutput<float>(),
124 m.GetOutput<float>(),
133 EXPECT_THAT(m.GetOutput<float>(),
144 EXPECT_THAT(m.GetOutput<float>(),
156 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({3, 4})))
[all...] |
| strided_slice_test.cc | 62 std::vector<input_type> GetOutput() {
98 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3}));
119 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3}));
130 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3}));
141 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2}));
152 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3, 4}));
163 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3}));
174 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3}));
185 EXPECT_THAT(m.GetOutput(), ElementsAreArray({4}));
196 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3, 2}))
[all...] |
| pack_test.cc | 47 std::vector<T> GetOutput() { return ExtractVector<T>(output_); }
62 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 4, 2, 5, 3, 6}));
72 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6}));
81 EXPECT_THAT(model.GetOutput(),
92 EXPECT_THAT(model.GetOutput(),
106 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 4, 2, 5, 3, 6}));
116 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6}));
125 EXPECT_THAT(model.GetOutput(),
137 EXPECT_THAT(model.GetOutput(),
148 EXPECT_THAT(model.GetOutput(),
[all...] |
| neg_test.cc | 42 std::vector<T> GetOutput() {
55 EXPECT_THAT(m.GetOutput<float>(),
63 EXPECT_THAT(m.GetOutput<int32_t>(), ElementsAreArray({2, 1, 0, -1, -2, -3}));
70 EXPECT_THAT(m.GetOutput<int64_t>(), ElementsAreArray({2, 1, 0, -1, -2, -3}));
|
| l2norm_test.cc | 54 std::vector<T> GetOutput() {
76 EXPECT_THAT(m.GetOutput<float>(),
89 EXPECT_THAT(m.GetOutput<float>(),
102 EXPECT_THAT(m.GetOutput<uint8_t>(),
114 EXPECT_THAT(m.GetOutput<int8_t>(),
132 EXPECT_THAT(m.GetOutput<uint8_t>(),
158 EXPECT_THAT(m.GetOutput<int8_t>(), ElementsAreArray({
|
| div_test.cc | 52 std::vector<float> GetOutput() { return ExtractVector<float>(output_); }
59 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); }
69 EXPECT_THAT(m.GetOutput(),
80 EXPECT_THAT(m.GetOutput(),
95 m.GetOutput(),
112 m.GetOutput(),
125 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, -1, 5, 1}));
135 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-1, 1, 0, 1}));
148 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-20, 1, 0, 1, -1, 20}))
163 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-6, 7, 2, 2, 3, -41})
[all...] |
| one_hot_test.cc | 57 std::vector<T> GetOutput() { return ExtractVector<T>(output_); }
72 EXPECT_THAT(model.GetOutput(),
83 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 0, 0, 0, 1, 0, 0, 0, 1}));
93 EXPECT_THAT(model.GetOutput(),
105 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 0, 0}));
115 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 0, 0, 0, 0, 1, 0, 0}));
128 EXPECT_THAT(model.GetOutput(),
142 EXPECT_THAT(model.GetOutput(),
156 EXPECT_THAT(model.GetOutput(),
172 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 0, 0, 0, 1, 0, 0, 0, 1}))
[all...] |
| squeeze_test.cc | 55 std::vector<float> GetOutput() { return ExtractVector<float>(output_); }
69 m.GetOutput(),
85 m.GetOutput(),
101 m.GetOutput(),
114 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3.85}));
|
| mirror_pad_test.cc | 44 std::vector<T> GetOutput() { return ExtractVector<T>(output_id_); }
59 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6}));
69 EXPECT_THAT(model.GetOutput(),
80 EXPECT_THAT(model.GetOutput(),
91 EXPECT_THAT(model.GetOutput(),
102 EXPECT_THAT(model.GetOutput(),
113 EXPECT_THAT(model.GetOutput(),
125 EXPECT_THAT(model.GetOutput(),
138 model.GetOutput(),
151 EXPECT_THAT(model.GetOutput(),
[all...] |
| select_test.cc | 46 std::vector<T> GetOutput() {
68 EXPECT_THAT(model.GetOutput<bool>(),
82 EXPECT_THAT(model.GetOutput<float>(), ElementsAreArray({0.1, 0.6, 0.3, 0.8}));
95 EXPECT_THAT(model.GetOutput<uint8_t>(), ElementsAreArray({5, 2, 7, 8}));
108 EXPECT_THAT(model.GetOutput<int8_t>(), ElementsAreArray({5, -2, 7, -8}));
121 EXPECT_THAT(model.GetOutput<int16_t>(), ElementsAreArray({5, 2, 7, 8}));
134 EXPECT_THAT(model.GetOutput<int32_t>(), ElementsAreArray({5, 2, 7, 8}));
146 EXPECT_THAT(model.GetOutput<int32_t>(), ElementsAreArray({5, 6, 3, 4}));
158 EXPECT_THAT(model.GetOutput<int32_t>(), ElementsAreArray({5, 6, 7, 8}));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| resize_nearest_neighbor_op_test.cc | 87 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
105 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
125 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
146 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
167 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
189 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
208 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
232 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
257 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
279 test::ExpectTensorEqual<float>(expected, *GetOutput(0))
[all...] |
| quantized_reshape_op_test.cc | 64 EXPECT_EQ(-10, GetOutput(1)->flat<float>()(0));
65 EXPECT_EQ(20, GetOutput(2)->flat<float>()(0));
66 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0));
|
| ragged_tensor_to_sparse_kernel_test.cc | 66 *GetOutput(kSparseIndicesOutput),
68 test::ExpectTensorEqual<int>(*GetOutput(kSparseValuesOutput),
70 test::ExpectTensorEqual<int64>(*GetOutput(kSparseDenseShapeOutput),
82 *GetOutput(kSparseIndicesOutput),
84 test::ExpectTensorEqual<int>(*GetOutput(kSparseValuesOutput),
86 test::ExpectTensorEqual<int64>(*GetOutput(kSparseDenseShapeOutput),
102 *GetOutput(kSparseIndicesOutput),
104 test::ExpectTensorEqual<int>(*GetOutput(kSparseValuesOutput),
106 test::ExpectTensorEqual<int64>(*GetOutput(kSparseDenseShapeOutput),
128 test::ExpectTensorEqual<int>(*GetOutput(kSparseValuesOutput)
[all...] |
| quantize_op_test.cc | 51 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0));
75 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0));
79 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1));
83 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2));
105 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0));
109 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1));
113 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2));
135 test::ExpectTensorEqual<qint8>(expected, *GetOutput(0));
139 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1));
143 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2))
[all...] |
| identity_n_op_test.cc | 47 test::ExpectTensorEqual<int32>(expected0, *GetOutput(0));
50 test::ExpectTensorEqual<double>(expected1, *GetOutput(1));
60 test::ExpectTensorEqual<int32>(expected, *GetOutput(0));
62 test::ExpectTensorEqual<int32>(expected, *GetOutput(1));
72 test::ExpectTensorEqual<string>(expected0, *GetOutput(0));
75 test::ExpectTensorEqual<int32>(expected1, *GetOutput(1));
|
| control_flow_ops_test.cc | 46 test::ExpectTensorEqual<int32>(expected, *GetOutput(0));
47 EXPECT_EQ(nullptr, GetOutput(1));
57 test::ExpectTensorEqual<int32>(expected, *GetOutput(1));
58 EXPECT_EQ(nullptr, GetOutput(0));
68 test::ExpectTensorEqual<int32>(expected, *GetOutput(0));
69 EXPECT_EQ(nullptr, GetOutput(1));
79 test::ExpectTensorEqual<string>(expected, *GetOutput(1));
80 EXPECT_EQ(nullptr, GetOutput(0));
|
| ragged_range_op_test.cc | 56 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput),
59 *GetOutput(kValuesOutput),
71 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput),
74 *GetOutput(kValuesOutput),
86 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput),
89 *GetOutput(kValuesOutput),
101 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput),
103 test::ExpectTensorEqual<int>(*GetOutput(kValuesOutput),
115 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput),
118 *GetOutput(kValuesOutput)
[all...] |
| fused_batch_norm_op_test.cc | 55 test::ExpectTensorNear<float>(expected, *GetOutput(0), 0.01);
59 test::ExpectTensorNear<float>(expected_mean, *GetOutput(1), 0.01);
63 test::ExpectTensorNear<float>(expected_variance, *GetOutput(2), 0.01);
89 test::ExpectTensorNear<float>(expected, *GetOutput(0), 0.01);
117 test::ExpectTensorNear<float>(expected_x, *GetOutput(0), 0.01);
121 test::ExpectTensorNear<float>(expected_scale, *GetOutput(1), 0.01);
125 test::ExpectTensorNear<float>(expected_offset, *GetOutput(2), 0.01);
|
