Home
last modified time | relevance | path

Searched refs:SetInput (Results 1 – 25 of 74) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/
Dpack_test.cc43 void SetInput(int index, std::initializer_list<T> data) { in SetInput() function in tflite::__anon908f57500111::PackOpModel
57 model.SetInput(0, {1, 4}); in TEST()
58 model.SetInput(1, {2, 5}); in TEST()
59 model.SetInput(2, {3, 6}); in TEST()
67 model.SetInput(0, {1, 4}); in TEST()
68 model.SetInput(1, {2, 5}); in TEST()
69 model.SetInput(2, {3, 6}); in TEST()
77 model.SetInput(0, {1, 2, 3, 4, 5, 6}); in TEST()
78 model.SetInput(1, {7, 8, 9, 10, 11, 12}); in TEST()
87 model.SetInput(0, {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST()
[all …]
Dconcatenation_test.cc55 void SetInput(int index, std::initializer_list<float> data) { in SetInput() function in tflite::__anon0d1bc6570111::ConcatenationOpModel
82 void SetInput(int index, std::initializer_list<float> data) { in SetInput() function in tflite::__anon0d1bc6570111::QuantizedConcatenationOpModel
99 m0.SetInput(0, {1.0f, 3.0f, 4.0f, 7.0f}); in TEST()
107 m0.SetInput(0, {5.0f}); in TEST()
115 m0.SetInput(0, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f}); in TEST()
127 m0.SetInput(0, tensor0); in TEST()
128 m0.SetInput(1, tensor1); in TEST()
135 m0_negative.SetInput(0, tensor0); in TEST()
136 m0_negative.SetInput(1, tensor1); in TEST()
143 m1.SetInput(0, tensor0); in TEST()
[all …]
Dresize_bilinear_test.cc50 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anond3a61b480111::ResizeBilinearOpModel
68 m.SetInput<float>({3, 6}); in TEST()
75 const_m.SetInput<float>({3, 6}); in TEST()
83 m.SetInput<uint8>({3, 6}); in TEST()
90 const_m.SetInput<uint8>({3, 6}); in TEST()
98 m.SetInput<int8_t>({3, 6}); in TEST()
105 const_m.SetInput<int8_t>({3, 6}); in TEST()
113 m.SetInput<float>({3, 9}); in TEST()
120 const_m.SetInput<float>({3, 9}); in TEST()
128 m.SetInput<uint8>({3, 9}); in TEST()
[all …]
Dresize_nearest_neighbor_test.cc50 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon6a3f65c10111::ResizeNearestNeighborOpModel
68 m.SetInput<float>({3, 6}); in TEST()
76 const_m.SetInput<float>({3, 6}); in TEST()
84 m.SetInput<uint8>({3, 6}); in TEST()
92 const_m.SetInput<uint8>({3, 6}); in TEST()
100 m.SetInput<int8_t>({-3, 6}); in TEST()
107 const_m.SetInput<int8_t>({-3, 6}); in TEST()
115 m.SetInput<float>({3, 9}); in TEST()
123 const_m.SetInput<float>({3, 9}); in TEST()
131 m.SetInput<uint8>({3, 9}); in TEST()
[all …]
Dactivations_test.cc76 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anonabb0f0760111::FloatActivationsOpModel
106 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anonabb0f0760111::QuantizedActivationsOpModel
124 m.SetInput({ in TEST()
138 m.SetInput({ in TEST()
152 m.SetInput({ in TEST()
166 m.SetInput({ in TEST()
180 m.SetInput({ in TEST()
198 m.SetInput<uint8_t>({ in TEST()
221 m.SetInput<int8_t>({ in TEST()
243 m.SetInput<uint8_t>({ in TEST()
[all …]
Dstrided_slice_test.cc49 void SetInput(std::initializer_list<input_type> data) { in SetInput() function in tflite::__anon1c38552f0111::StridedSliceOpModel
92 m.SetInput({1, 2, 3, 4}); in TEST()
103 m.SetInput({1, 2, 3, 4}); in TEST()
113 m.SetInput({1, 2, 3, 4}); in TEST()
124 m.SetInput({1, 2, 3, 4}); in TEST()
135 m.SetInput({1, 2, 3, 4}); in TEST()
146 m.SetInput({1, 2, 3, 4}); in TEST()
157 m.SetInput({1, 2, 3, 4}); in TEST()
168 m.SetInput({1, 2, 3, 4}); in TEST()
179 m.SetInput({1, 2, 3, 4}); in TEST()
[all …]
Dgather_nd_test.cc40 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon4a1df51d0111::GatherNdOpModel
64 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST()
73 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST()
83 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST()
93 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST()
102 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST()
112 m.SetInput<float>({1.1, -1.2, 1.3, -2.1, 2.2, 2.3, // in TEST()
124 m.SetInput<float>({1.1, -1.2, 1.3, -2.1, 2.2, 2.3, // in TEST()
138 m.SetInput<float>({1.1, -1.2, 1.3, -2.1, 2.2, 2.3, // in TEST()
150 m.SetInput<float>({1.1, -1.2, 1.3, -2.1, 2.2, 2.3, // in TEST()
[all …]
Dgather_test.cc40 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon6134f40c0111::GatherOpModel
71 m.SetInput<float>({-2.0, 0.2, 0.7, 0.8}); in TEST()
80 m.SetInput<float>({-2.0, 0.2, 0.7, 0.8}); in TEST()
92 m.SetInput<float>({1.0, 2.0, 3.0}); in TEST()
101 m.SetInput<float>({1.0, 3.0, 5.0}); in TEST()
110 m.SetInput<float>({1.0, 2.0, 3.0}); in TEST()
120 m.SetInput<float>({-2.0, 0.2, 0.7, 0.8}); in TEST()
130 m.SetInput<float>({-2.0, 0.2, 0.7, 0.8}); in TEST()
141 m.SetInput<float>({1, 2, 3, 4, 5, 6}); in TEST()
153 m.SetInput<float>({1, 2, 3, 4, 5, 6}); in TEST()
[all …]
Dskip_gram_test.cc44 void SetInput(const string& content) { in SetInput() function in tflite::__anon4f60efae0111::SkipGramOp
68 m.SetInput(kSentence); in TEST()
77 m.SetInput(kSentence); in TEST()
87 m.SetInput(kSentence); in TEST()
101 m.SetInput(kSentence); in TEST()
119 m.SetInput(kSentence); in TEST()
131 m.SetInput(kSentence); in TEST()
145 m.SetInput(kSentence); in TEST()
161 m.SetInput(kSentence); in TEST()
184 m.SetInput(kSentence); in TEST()
[all …]
Dpooling_test.cc55 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon63a61fa90111::FloatPoolingOpModel
66 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon63a61fa90111::QuantizedPoolingOpModel
70 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon63a61fa90111::QuantizedPoolingOpModel
85 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon63a61fa90111::SymmetricQuantizedPoolingOpModel
89 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon63a61fa90111::SymmetricQuantizedPoolingOpModel
123 m.SetInput({ in TEST()
139 m.SetInput({ in TEST()
161 m.SetInput(input); in TEST()
184 m.SetInput(ReplicateDepthRamp(input_image_plane, depth, 1.f / 512.f)); in TEST()
205 m.SetInput({ in TEST()
[all …]
Dconv_test.cc120 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon8d129a670111::ConvolutionOpModel
146 m.SetInput({ in TEST_P()
181 m.SetInput({ in TEST_P()
211 m.SetInput({ in TEST_P()
235 m.SetInput({ in TEST_P()
268 m.SetInput({ in TEST_P()
298 m.SetInput({ in TEST_P()
336 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST_P()
402 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST_P()
457 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST_P()
[all …]
Dslice_test.cc42 void SetInput(std::initializer_list<input_type> data) { in SetInput() function in tflite::__anon667d06810111::SliceOpModel
67 m.SetInput({1, 2, 3, 4}); in TEST()
78 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
89 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST()
101 m.SetInput({1, 2, 3, 4}); in TEST()
112 m.SetInput({1, 2, 3, 4}); in TEST()
125 m.SetInput({1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6}); in TEST()
136 m.SetInput({1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6}); in TEST()
147 m.SetInput({1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6}); in TEST()
158 m.SetInput({1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6}); in TEST()
[all …]
Dtile_test.cc38 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anone8b6eaff0111::TileOpModel
61 m.SetInput<float>({1.f, 2.f, 3.f}); in TEST()
70 m.SetInput<float>({ in TEST()
99 m.SetInput<float>({ in TEST()
120 m.SetInput<uint8_t>({ in TEST()
149 m.SetInput<int32_t>({ in TEST()
178 m.SetInput<bool>({true, false, false, true, true, false}); in TEST()
191 m.SetInput<int64_t>({ in TEST()
220 m.SetInput<int64_t>({ in TEST()
Dunidirectional_sequence_rnn_test.cc218 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon2f8e60700111::UnidirectionalRNNOpModel
222 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon2f8e60700111::UnidirectionalRNNOpModel
286 rnn.SetInput(0, batch_start, batch_end); in TEST()
287 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
311 rnn.SetInput(0, batch_start, batch_end); in TEST()
312 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
337 rnn.SetInput(0, batch_start, batch_end); in TEST()
338 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
364 rnn.SetInput(2 * i * rnn.input_size(), batch_start, batch_end); in TEST()
365 rnn.SetInput((2 * i + 1) * rnn.input_size(), batch_start, batch_end); in TEST()
[all …]
Dtranspose_test.cc153 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anona7b28c900111::TransposeOpModel
225 m.SetInput({1, 2, 3}); in TEST()
233 m.SetInput({1, 2, 3}); in TEST()
242 m.SetInput({0, 1, 2, 3, 4, 5}); in TEST()
250 m.SetInput({0, 1, 2, 3, 4, 5}); in TEST()
259 m.SetInput({0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST()
270 m.SetInput({0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST()
289 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
297 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
306 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
[all …]
Ddepthwise_conv_test.cc125 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anona84de9d30111::DepthwiseConvolutionOpModel
152 m.SetInput({ in TEST_P()
199 m.SetInput({0, 0, 0, 0, 0, 0, 0, 0, 0, in TEST_P()
245 m.SetInput({1, 1, 1, 1, 1, 1, 1, 1, 1}); in TEST_P()
266 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anona84de9d30111::QuantizedDepthwiseConvolutionOpModel
300 m.SetInput({ in TEST_P()
352 quant_op.SetInput(input); in TEST_P()
357 float_op.SetInput(input); in TEST_P()
397 m.SetInput({0, 0, 0, 0, 0, 0, 0, 0, 0, in TEST_P()
449 m.SetInput({1, 1, 1, 1, 1, 1, 1, 1, 1}); in TEST_P()
[all …]
Dembedding_lookup_test.cc47 void SetInput(std::initializer_list<int> data) { in SetInput() function in tflite::__anon9fb92c890111::BaseEmbeddingLookupOpModel
98 m.SetInput({1, 0, 2}); in TEST()
114 m.SetInput({1, 0, 2}); in TEST()
135 m.SetInput({1, 0, 2}); in TEST()
156 m.SetInput({1, 0, 2}); in TEST()
177 m.SetInput({1, 0, 2}); in TEST()
198 m.SetInput({1, 0, 2}); in TEST()
219 m.SetInput({1, 0, 2}); in TEST()
Dfully_connected_test.cc206 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon3e7c6b6f0111::FloatFullyConnectedOpModel
209 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon3e7c6b6f0111::FloatFullyConnectedOpModel
256 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon3e7c6b6f0111::QuantizedFullyConnectedOpModel
312 void SetInput(const std::vector<float>& f) { PopulateTensor(input_, f); } in SetInput() function in tflite::__anon3e7c6b6f0111::HybridFullyConnectedOpModel
382 m.SetInput({ in TEST_P()
400 m.SetInput({ in TEST_P()
421 m.SetInput({ in TEST()
445 m.SetInput<uint8_t>({ in TEST_P()
475 m.SetInput<int8_t>({ in TEST_P()
502 m.SetInput<uint8_t>({ in TEST_P()
[all …]
Dspace_to_depth_test.cc39 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon5bb25c730111::SpaceToDepthOpModel
62 m.SetInput<float>({1.4, 2.3, 3.2, 4.1, 5.4, 6.3, 7.2, 8.1}); in TEST()
71 m.SetInput<uint8_t>({1, 2, 3, 4}); in TEST()
79 m.SetInput<int8_t>({1, 2, 3, 4}); in TEST()
87 m.SetInput<int32_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST()
96 m.SetInput<int64_t>({1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16}); in TEST()
Dunpack_test.cc44 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon05b8f2d30111::UnpackOpModel
73 model.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
93 model.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
111 model.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
129 model.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
149 model.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
165 model.SetInput({1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
184 model.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
205 model.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
224 model.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
[all …]
Dbidirectional_sequence_rnn_test.cc739 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon65c37d6f0111::BidirectionalRNNOpModel
743 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon65c37d6f0111::BidirectionalRNNOpModel
801 rnn.SetInput(0, batch_start, batch_end); in TEST()
802 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
844 rnn.SetInput(2 * i * rnn.input_size(), batch_start, batch_end); in TEST()
845 rnn.SetInput((2 * i + 1) * rnn.input_size(), batch_start, batch_end); in TEST()
877 rnn.SetInput(0, batch_start, batch_end); in TEST()
878 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
920 rnn.SetInput(2 * i * rnn.input_size(), batch_start, batch_end); in TEST()
921 rnn.SetInput((2 * i + 1) * rnn.input_size(), batch_start, batch_end); in TEST()
[all …]
Dlocal_response_norm_test.cc40 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon8a74d9650111::LocalResponseNormOpModel
54 m.SetInput({-1.1, 0.6, 0.7, 1.2, -0.7, 0.1}); in TEST()
65 m.SetInput({-1.1, 0.6, 0.7, 1.2, -0.7, 0.1}); in TEST()
75 m.SetInput({-1.1, 0.6, 0.7, 1.2, -0.7, 0.1}); in TEST()
86 m.SetInput({-1.1, 0.6, 0.7, 1.2, -0.7, 0.1}); in TEST()
Dreshape_test.cc68 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anonc79801990111::ReshapeOpModel
130 m.SetInput({3}); in TEST_P()
164 m.SetInput({5, 6, 7, 8}); in TEST_P()
173 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8}); in TEST_P()
181 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8}); in TEST_P()
191 m.SetInput({3}); in TEST_P()
208 m.SetInput({3}); in TEST_P()
213 m.SetInput({3}); in TEST_P()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc213 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon6d004b400111::FloatPoolingOpModel
229 m.SetInput({ in TEST()
242 m.SetInput({ in TEST()
255 m.SetInput({ in TEST()
305 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon6d004b400111::ConvolutionOpModel
352 m.SetInput({ in TEST()
390 m.SetInput({ in TEST()
422 m.SetInput({ in TEST()
486 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon6d004b400111::DepthwiseConvolutionOpModel
504 m.SetInput({ in TEST()
[all …]
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver_test.cc43 runner->SetInput(0, "0.1,0.2,0.3,0.4"); in TEST()
44 runner->SetInput(1, "0.001,0.002,0.003,0.004"); in TEST()
45 runner->SetInput(2, "0.001,0.002,0.003,0.004"); in TEST()
46 runner->SetInput(3, "0.01,0.02,0.03,0.04"); in TEST()
79 runner->SetInput(0, "0.1,0.2,0.3,0.4"); in TEST()
80 runner->SetInput(1, "0.001,0.002,0.003,0.004"); in TEST()
81 runner->SetInput(2, "0.001,0.002,0.003,0.004"); in TEST()
82 runner->SetInput(3, "0.01,0.02,0.03,0.04"); in TEST()

123