Home
last modified time | relevance | path

Searched refs:GetOutput (Results 1 – 25 of 245) sorted by relevance

12345678910

/external/tensorflow/tensorflow/lite/kernels/
Dresize_bilinear_test.cc56 std::vector<T> GetOutput() { in GetOutput() function in tflite::__anond3a61b480111::ResizeBilinearOpModel
71 EXPECT_THAT(m.GetOutput<float>(), in TEST()
77 EXPECT_THAT(const_m.GetOutput<float>(), in TEST()
86 EXPECT_THAT(m.GetOutput<uint8>(), in TEST()
92 EXPECT_THAT(const_m.GetOutput<uint8>(), in TEST()
101 EXPECT_THAT(m.GetOutput<int8_t>(), in TEST()
107 EXPECT_THAT(const_m.GetOutput<int8_t>(), in TEST()
116 EXPECT_THAT(m.GetOutput<float>(), in TEST()
122 EXPECT_THAT(const_m.GetOutput<float>(), in TEST()
131 EXPECT_THAT(m.GetOutput<uint8>(), in TEST()
[all …]
Dresize_nearest_neighbor_test.cc56 std::vector<T> GetOutput() { in GetOutput() function in tflite::__anon6a3f65c10111::ResizeNearestNeighborOpModel
71 EXPECT_THAT(m.GetOutput<float>(), in TEST()
78 EXPECT_THAT(const_m.GetOutput<float>(), in TEST()
87 EXPECT_THAT(m.GetOutput<uint8>(), in TEST()
94 EXPECT_THAT(const_m.GetOutput<uint8>(), in TEST()
103 EXPECT_THAT(m.GetOutput<int8_t>(), in TEST()
109 EXPECT_THAT(const_m.GetOutput<int8_t>(), in TEST()
118 EXPECT_THAT(m.GetOutput<float>(), in TEST()
125 EXPECT_THAT(const_m.GetOutput<float>(), in TEST()
134 EXPECT_THAT(m.GetOutput<uint8>(), in TEST()
[all …]
Dstrided_slice_test.cc62 std::vector<input_type> GetOutput() { in GetOutput() function in tflite::__anon1c38552f0111::StridedSliceOpModel
98 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3})); in TEST()
119 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3})); in TEST()
130 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3})); in TEST()
141 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2})); in TEST()
152 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3, 4})); in TEST()
163 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3})); in TEST()
174 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3})); in TEST()
185 EXPECT_THAT(m.GetOutput(), ElementsAreArray({4})); in TEST()
196 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3, 2})); in TEST()
[all …]
Dgather_nd_test.cc50 std::vector<T> GetOutput() { in GetOutput() function in tflite::__anon4a1df51d0111::GatherNdOpModel
68 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 2.2})); in TEST()
77 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({2.1, 2.2, 1.1, 1.2})); in TEST()
87 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({2.1, 2.2, 1.1, 1.2})); in TEST()
97 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 2.2})); in TEST()
106 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 1.1})); in TEST()
118 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({-1.2, -4.1})); in TEST()
130 EXPECT_THAT(m.GetOutput<float>(), in TEST()
144 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({-1.2, -4.1})); in TEST()
156 EXPECT_THAT(m.GetOutput<float>(), in TEST()
[all …]
Dgather_test.cc54 std::vector<T> GetOutput() { in GetOutput() function in tflite::__anon6134f40c0111::GatherOpModel
74 EXPECT_THAT(m.GetOutput<float>(), in TEST()
83 EXPECT_THAT(m.GetOutput<float>(), in TEST()
95 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({2.0}))); in TEST()
104 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({3.0}))); in TEST()
113 EXPECT_THAT(m.GetOutput<float>(), in TEST()
124 m.GetOutput<float>(), in TEST()
133 EXPECT_THAT(m.GetOutput<float>(), in TEST()
144 EXPECT_THAT(m.GetOutput<float>(), in TEST()
156 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({3, 4}))); in TEST()
[all …]
Dskip_gram_test.cc48 std::vector<string> GetOutput() { in GetOutput() function in tflite::__anon4f60efae0111::SkipGramOp
70 EXPECT_THAT(m.GetOutput(), testing::UnorderedElementsAreArray( in TEST()
79 EXPECT_THAT(m.GetOutput(), in TEST()
89 EXPECT_THAT(m.GetOutput(), in TEST()
103 EXPECT_THAT(m.GetOutput(), in TEST()
122 m.GetOutput(), in TEST()
133 EXPECT_THAT(m.GetOutput(), in TEST()
147 EXPECT_THAT(m.GetOutput(), in TEST()
163 EXPECT_THAT(m.GetOutput(), in TEST()
187 m.GetOutput(), in TEST()
[all …]
Dcomparisons_test.cc50 std::vector<bool> GetOutput() { return ExtractVector<bool>(output_); } in GetOutput() function in tflite::__anon92c241bf0111::ComparisonOpModel
102 EXPECT_THAT(model.GetOutput(), ElementsAre(true, false, false, true)); in TEST()
113 EXPECT_THAT(model.GetOutput(), ElementsAre(true, false, false, false)); in TEST()
124 EXPECT_THAT(model.GetOutput(), ElementsAre(false, false, true, false)); in TEST()
135 EXPECT_THAT(model.GetOutput(), ElementsAre(false, false, true, false)); in TEST()
146 EXPECT_THAT(model.GetOutput(), ElementsAre(false, false, false, false, false, in TEST()
158 EXPECT_THAT(model.GetOutput(), ElementsAre(false, true, true, false)); in TEST()
169 EXPECT_THAT(model.GetOutput(), ElementsAre(false, true, true, true)); in TEST()
180 EXPECT_THAT(model.GetOutput(), ElementsAre(true, true, false, true)); in TEST()
191 EXPECT_THAT(model.GetOutput(), ElementsAre(true, true, false, true)); in TEST()
[all …]
Dpack_test.cc47 std::vector<T> GetOutput() { return ExtractVector<T>(output_); } in GetOutput() function in tflite::__anon908f57500111::PackOpModel
62 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 4, 2, 5, 3, 6})); in TEST()
72 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); in TEST()
81 EXPECT_THAT(model.GetOutput(), in TEST()
92 EXPECT_THAT(model.GetOutput(), in TEST()
106 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 4, 2, 5, 3, 6})); in TEST()
116 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); in TEST()
125 EXPECT_THAT(model.GetOutput(), in TEST()
137 EXPECT_THAT(model.GetOutput(), in TEST()
148 EXPECT_THAT(model.GetOutput(), in TEST()
[all …]
Dactivations_test.cc79 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anonabb0f0760111::FloatActivationsOpModel
111 std::vector<T> GetOutput() { in GetOutput() function in tflite::__anonabb0f0760111::QuantizedActivationsOpModel
129 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST()
143 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST()
157 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST()
171 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST()
185 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST()
210 EXPECT_THAT(m.GetOutput<uint8_t>(), in TEST()
232 EXPECT_THAT(m.GetOutput<int8_t>(), in TEST()
255 EXPECT_THAT(m.GetOutput<uint8_t>(), in TEST()
[all …]
Dmirror_pad_test.cc44 std::vector<T> GetOutput() { return ExtractVector<T>(output_id_); } in GetOutput() function in tflite::__anon97c8e8c00111::BaseMirrorPadOpModel
59 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); in TEST()
69 EXPECT_THAT(model.GetOutput(), in TEST()
80 EXPECT_THAT(model.GetOutput(), in TEST()
91 EXPECT_THAT(model.GetOutput(), in TEST()
102 EXPECT_THAT(model.GetOutput(), in TEST()
113 EXPECT_THAT(model.GetOutput(), in TEST()
125 EXPECT_THAT(model.GetOutput(), in TEST()
138 model.GetOutput(), in TEST()
151 EXPECT_THAT(model.GetOutput(), in TEST()
[all …]
Dwhere_test.cc49 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput() function in tflite::__anon075762ec0111::IntegerWhereOpModel
56 EXPECT_THAT(m.GetOutput().size(), 0); in TEST()
63 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2})); in TEST()
72 EXPECT_EQ(m.GetOutput().size(), 0); in TEST()
79 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, // in TEST()
89 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, // in TEST()
102 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, // in TEST()
116 EXPECT_EQ(m.GetOutput().size(), 0); in TEST()
124 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, // in TEST()
134 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, // in TEST()
[all …]
Dconv_test.cc123 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon8d129a670111::ConvolutionOpModel
163 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST_P()
198 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST_P()
227 EXPECT_THAT(m.GetOutput(), ElementsAreArray({10, 34})); in TEST_P()
251 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST_P()
285 EXPECT_THAT(m.GetOutput(), in TEST_P()
309 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST_P()
365 EXPECT_THAT(m.GetOutput(), ElementsAreArray({105, 150, 183, 95, 235, 312, 357, in TEST_P()
374 EXPECT_THAT(m.GetOutput(), in TEST_P()
431 EXPECT_THAT(m.GetOutput(), ElementsAreArray({115, 160, 193, 105, 245, 322, in TEST_P()
[all …]
Dconcatenation_test.cc58 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon0d1bc6570111::ConcatenationOpModel
86 std::vector<T> GetOutput() { in GetOutput() function in tflite::__anon0d1bc6570111::QuantizedConcatenationOpModel
101 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 3, 4, 7})); in TEST()
109 EXPECT_THAT(m0.GetOutput(), ::testing::ElementsAre(5)); in TEST()
117 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); in TEST()
130 EXPECT_THAT(m0.GetOutput(), in TEST()
138 EXPECT_THAT(m0_negative.GetOutput(), in TEST()
146 EXPECT_THAT(m1.GetOutput(), in TEST()
154 EXPECT_THAT(m1_negative.GetOutput(), in TEST()
166 EXPECT_THAT(m0.GetOutput(), in TEST()
[all …]
Dslice_test.cc52 std::vector<input_type> GetOutput() { in GetOutput() function in tflite::__anon667d06810111::SliceOpModel
72 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3})); in TEST()
83 EXPECT_THAT(m.GetOutput(), ElementsAreArray({4, 5})); in TEST()
94 EXPECT_THAT(m.GetOutput(), in TEST()
106 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3, 4})); in TEST()
117 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3, 4})); in TEST()
130 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3, 3, 3})); in TEST()
141 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3, 3, 3, 4, 4, 4})); in TEST()
152 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3, 3, 3, 5, 5, 5})); in TEST()
163 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3, 3, 3, 5, 5, 5})); in TEST()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dnon_max_suppression_op_test.cc57 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
71 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
86 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
103 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
118 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
130 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
153 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
194 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
227 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
243 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F()
[all …]
Dquantize_op_test.cc51 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0)); in TEST_F()
75 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0)); in TEST_F()
79 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1)); in TEST_F()
83 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2)); in TEST_F()
105 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0)); in TEST_F()
109 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1)); in TEST_F()
113 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2)); in TEST_F()
135 test::ExpectTensorEqual<qint8>(expected, *GetOutput(0)); in TEST_F()
139 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1)); in TEST_F()
143 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2)); in TEST_F()
[all …]
Dfake_quant_ops_test.cc71 Tensor* output = GetOutput(0); in RunTestFakeQuantWithMinMaxArgs()
100 Tensor* output = GetOutput(0); in RunTestFakeQuantWithMinMaxVars()
130 Tensor* output = GetOutput(0); in RunTestFakeQuantWithMinMaxVarsPerChannel()
378 Tensor* output = GetOutput(0); in TEST_F()
408 Tensor* output = GetOutput(0); in TEST_F()
439 Tensor* output = GetOutput(0); in TEST_F()
470 Tensor* output = GetOutput(0); in TEST_F()
667 Tensor* output_bprop_wrt_input = GetOutput(0); in TEST_F()
675 Tensor* output_bprop_wrt_min = GetOutput(1); in TEST_F()
680 Tensor* output_bprop_wrt_max = GetOutput(2); in TEST_F()
[all …]
Dragged_range_op_test.cc56 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput), in TEST_F()
59 *GetOutput(kValuesOutput), in TEST_F()
71 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput), in TEST_F()
74 *GetOutput(kValuesOutput), in TEST_F()
86 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput), in TEST_F()
89 *GetOutput(kValuesOutput), in TEST_F()
101 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput), in TEST_F()
103 test::ExpectTensorEqual<int>(*GetOutput(kValuesOutput), in TEST_F()
115 test::ExpectTensorEqual<int64>(*GetOutput(kSplitsOutput), in TEST_F()
118 *GetOutput(kValuesOutput), in TEST_F()
[all …]
Dresize_nearest_neighbor_op_test.cc87 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
105 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
125 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
146 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
167 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
189 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
208 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
232 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
257 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
279 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
[all …]
Dragged_tensor_to_sparse_kernel_test.cc66 *GetOutput(kSparseIndicesOutput), in TEST_F()
68 test::ExpectTensorEqual<int>(*GetOutput(kSparseValuesOutput), in TEST_F()
70 test::ExpectTensorEqual<int64>(*GetOutput(kSparseDenseShapeOutput), in TEST_F()
82 *GetOutput(kSparseIndicesOutput), in TEST_F()
84 test::ExpectTensorEqual<int>(*GetOutput(kSparseValuesOutput), in TEST_F()
86 test::ExpectTensorEqual<int64>(*GetOutput(kSparseDenseShapeOutput), in TEST_F()
102 *GetOutput(kSparseIndicesOutput), in TEST_F()
104 test::ExpectTensorEqual<int>(*GetOutput(kSparseValuesOutput), in TEST_F()
106 test::ExpectTensorEqual<int64>(*GetOutput(kSparseDenseShapeOutput), in TEST_F()
128 test::ExpectTensorEqual<int>(*GetOutput(kSparseValuesOutput), in TEST_F()
[all …]
Dragged_gather_op_test.cc77 test::ExpectTensorEqual<int64>(*GetOutput(0), in TEST_F()
80 *GetOutput(1), in TEST_F()
99 test::ExpectTensorEqual<int64>(*GetOutput(0), in TEST_F()
101 test::ExpectTensorEqual<int64>(*GetOutput(1), in TEST_F()
104 *GetOutput(2), test::AsTensor<float>({.1, .2, .3, .4, .5, .6, .7, .8}), in TEST_F()
126 test::ExpectTensorEqual<int64>(*GetOutput(0), in TEST_F()
128 test::ExpectTensorEqual<int64>(*GetOutput(1), in TEST_F()
131 *GetOutput(2), in TEST_F()
150 test::ExpectTensorEqual<int64>(*GetOutput(0), in TEST_F()
152 test::ExpectTensorEqual<int64>(*GetOutput(1), in TEST_F()
[all …]
Ddynamic_partition_op_test.cc66 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
71 test::ExpectTensorEqual<float>(expected, *GetOutput(1)); in TEST_F()
76 test::ExpectTensorEqual<float>(expected, *GetOutput(2)); in TEST_F()
81 test::ExpectTensorEqual<float>(expected, *GetOutput(3)); in TEST_F()
99 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
104 test::ExpectTensorEqual<float>(expected, *GetOutput(1)); in TEST_F()
109 test::ExpectTensorEqual<float>(expected, *GetOutput(2)); in TEST_F()
114 test::ExpectTensorEqual<float>(expected, *GetOutput(3)); in TEST_F()
134 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
137 test::ExpectTensorEqual<float>(expected_empty, *GetOutput(1)); in TEST_F()
[all …]
Dparse_tensor_test.cc83 ParseSerializedOutput<Eigen::half>(GetOutput(0), &parse_output); in TEST_F()
92 ParseSerializedOutput<float>(GetOutput(0), &parse_output); in TEST_F()
101 ParseSerializedOutput<double>(GetOutput(0), &parse_output); in TEST_F()
110 ParseSerializedOutput<int64>(GetOutput(0), &parse_output); in TEST_F()
119 ParseSerializedOutput<int32>(GetOutput(0), &parse_output); in TEST_F()
128 ParseSerializedOutput<int16>(GetOutput(0), &parse_output); in TEST_F()
137 ParseSerializedOutput<int8>(GetOutput(0), &parse_output); in TEST_F()
146 ParseSerializedOutput<uint16>(GetOutput(0), &parse_output); in TEST_F()
155 ParseSerializedOutput<uint8>(GetOutput(0), &parse_output); in TEST_F()
165 ParseSerializedOutput<complex64>(GetOutput(0), &parse_output); in TEST_F()
[all …]
Dcolorspace_op_test.cc48 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckBlack()
58 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckGray()
68 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckWhite()
82 test::ExpectTensorNear<T>(expected, *GetOutput(0), 1e-6); in CheckRedMax()
96 test::ExpectTensorNear<T>(expected, *GetOutput(0), 1e-6); in CheckGreenMax()
110 test::ExpectTensorNear<T>(expected, *GetOutput(0), 1e-6); in CheckBlueMax()
123 test::ExpectTensorNear<T>(expected, *GetOutput(0), 1e-6); in CheckNegativeDifference()
144 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckBlack()
154 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckGray()
164 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckWhite()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc107 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon6d004b400111::FloatAddOpModel
123 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-1.9, 0.4, 1.0, 1.3})); in TEST()
136 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-1.0, 1.0, 4.0, 6.0})); in TEST()
147 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0.0, 0.4, 1.0, 1.3})); in TEST()
177 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon6d004b400111::FloatMulOpModel
192 EXPECT_THAT(m.GetOutput(), in TEST()
217 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon6d004b400111::FloatPoolingOpModel
234 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2.75, 5.75})); in TEST()
247 EXPECT_THAT(m.GetOutput(), ElementsAreArray({6, 10})); in TEST()
260 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3.5, 6.5})); in TEST()
[all …]

12345678910