Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dconcatenation_test.cc54 void SetInput(int index, std::initializer_list<float> data) { in SetInput() function in tflite::__anon4d3a9c4b0111::ConcatenationOpModel
63 void SetInput(int index, std::initializer_list<float> data) { in SetInput() function in tflite::__anon4d3a9c4b0111::QuantizedConcatenationOpModel
76 m0.SetInput(0, {1.0f, 3.0f, 4.0f, 7.0f}); in TEST()
84 m0.SetInput(0, {5.0f}); in TEST()
92 m0.SetInput(0, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f}); in TEST()
104 m0.SetInput(0, tensor0); in TEST()
105 m0.SetInput(1, tensor1); in TEST()
112 m0_negative.SetInput(0, tensor0); in TEST()
113 m0_negative.SetInput(1, tensor1); in TEST()
120 m1.SetInput(0, tensor0); in TEST()
[all …]
Dstrided_slice_test.cc48 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon5c572b230111::StridedSliceOpModel
87 m.SetInput({1, 2, 3, 4}); in TEST()
98 m.SetInput({1, 2, 3, 4}); in TEST()
108 m.SetInput({1, 2, 3, 4}); in TEST()
119 m.SetInput({1, 2, 3, 4}); in TEST()
130 m.SetInput({1, 2, 3, 4}); in TEST()
141 m.SetInput({1, 2, 3, 4}); in TEST()
152 m.SetInput({1, 2, 3, 4}); in TEST()
163 m.SetInput({1, 2, 3, 4}); in TEST()
174 m.SetInput({1, 2, 3, 4}); in TEST()
[all …]
Dskip_gram_test.cc44 void SetInput(const string& content) { in SetInput() function in tflite::__anon3bbaeba20111::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 …]
Dactivations_test.cc64 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon1679b96a0111::FloatActivationsOpModel
80 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon1679b96a0111::QuantizedActivationsOpModel
93 m.SetInput({ in TEST()
107 m.SetInput({ in TEST()
121 m.SetInput({ in TEST()
135 m.SetInput({ in TEST()
149 m.SetInput({ in TEST()
164 m.SetInput({ in TEST()
183 m.SetInput({ in TEST()
196 m2.SetInput({ in TEST()
[all …]
Dresize_bilinear_test.cc48 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon96d23e3c0111::ResizeBilinearOpModel
63 m.SetInput({3, 6}); in TEST()
69 const_m.SetInput({3, 6}); in TEST()
76 m.SetInput({3, 9}); in TEST()
82 const_m.SetInput({3, 9}); in TEST()
89 m.SetInput({ in TEST()
102 const_m.SetInput({ in TEST()
116 m.SetInput({ in TEST()
134 const_m.SetInput({ in TEST()
153 m.SetInput({ in TEST()
[all …]
Dspace_to_batch_nd_test.cc29 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon09a4ca450111::SpaceToBatchNDOpModel
104 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST()
113 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST()
124 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST()
133 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST()
144 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST()
155 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST()
168 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
180 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
Dtranspose_test.cc130 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon940c88840111::TransposeOpModel
200 m.SetInput({1, 2, 3}); in TEST()
208 m.SetInput({1, 2, 3}); in TEST()
217 m.SetInput({0, 1, 2, 3, 4, 5}); in TEST()
225 m.SetInput({0, 1, 2, 3, 4, 5}); in TEST()
234 m.SetInput({0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST()
245 m.SetInput({0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST()
262 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
270 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
279 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
[all …]
Dpooling_test.cc55 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anonc5a88e9d0111::FloatPoolingOpModel
66 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anonc5a88e9d0111::QuantizedPoolingOpModel
82 m.SetInput({ in TEST()
98 m.SetInput({ in TEST()
114 m.SetInput({ in TEST()
130 m.SetInput({ in TEST()
146 m.SetInput({ in TEST()
Dfully_connected_test.cc180 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon01a88e630111::FloatFullyConnectedOpModel
183 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon01a88e630111::FloatFullyConnectedOpModel
200 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon01a88e630111::QuantizedFullyConnectedOpModel
222 m.SetInput({ in TEST()
246 m.SetInput({ in TEST()
274 m.SetInput({ in TEST()
301 m.SetInput({ in TEST()
354 m.SetInput(0, batch_start, batch_end); in TEST()
355 m.SetInput(m.input_size(), batch_start, batch_end); in TEST()
Dconv_test.cc98 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon3a8e4ddb0111::ConvolutionOpModel
124 m.SetInput({ in TEST_P()
154 m.SetInput({ in TEST_P()
192 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST_P()
246 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST_P()
301 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST_P()
356 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST_P()
381 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon3a8e4ddb0111::QuantizedConvolutionOpModel
407 m.SetInput({ in TEST_P()
448 m.SetInput({ in TEST_P()
Dlocal_response_norm_test.cc40 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon7c2156590111::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()
Dspace_to_depth_test.cc39 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon9fabf0e70111::SpaceToDepthOpModel
60 m.SetInput<float>({1.4, 2.3, 3.2, 4.1, 5.4, 6.3, 7.2, 8.1}); in TEST()
69 m.SetInput<uint8_t>({1, 2, 3, 4}); in TEST()
77 m.SetInput<int32_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST()
86 m.SetInput<int64_t>({1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16}); in TEST()
Dsqueeze_test.cc51 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anondda50d070111::FloatSqueezeOpModel
65 m.SetInput(data); in TEST()
81 m.SetInput(data); in TEST()
97 m.SetInput(data); in TEST()
111 m.SetInput(data); in TEST()
Dsoftmax_test.cc43 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon09ef64c70111::SoftmaxOpModel
47 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon09ef64c70111::SoftmaxOpModel
64 m.SetInput({ in TEST()
90 m.SetInput(0, input_buffer, input_buffer + input_size * batch_size); in TEST()
118 m.SetInput(0, input_buffer, input_buffer + input_size * batch_size); in TEST()
Dpad_test.cc28 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anond79f939a0111::PadOpModel
109 m.SetInput({1, 2, 3, 4}); in TEST()
118 m.SetInput({1, 2, 3, 4}); in TEST()
128 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
138 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST()
Dunidirectional_sequence_rnn_test.cc167 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon5a9291640111::UnidirectionalRNNOpModel
171 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon5a9291640111::UnidirectionalRNNOpModel
259 rnn.SetInput(0, batch_start, batch_end); in TEST()
260 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
327 rnn.SetInput(2 * i * rnn.input_size(), batch_start, batch_end); in TEST()
328 rnn.SetInput((2 * i + 1) * rnn.input_size(), batch_start, batch_end); in TEST()
Dmean_test.cc31 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon1603f8660111::BaseMeanOpModel
83 m.SetInput(data); in TEST()
95 m.SetInput(data); in TEST()
111 m.SetInput(data); in TEST()
126 m.SetInput(data); in TEST()
Dembedding_lookup_sparse_test.cc52 void SetInput(std::initializer_list<int> lookup_data, in SetInput() function in tflite::__anone85c4d0a0111::EmbeddingLookupSparseOpModel
89 m.SetInput({1, 3, 0}, {0, 0, 2, 0, 2, 1}, {3, 2}, {1.0, 2.0, 4.0}); in TEST()
105 m.SetInput({1, 3, 0}, {0, 0, 2, 0, 2, 1}, {3, 2}, {1.0, 2.0, 4.0}); in TEST()
121 m.SetInput({1, 3, 0}, {0, 0, 2, 0, 2, 1}, {3, 2}, {1.0, 2.0, 4.0}); in TEST()
140 m.SetInput({1, 3, 0}, {0, 0, 0, 2, 0, 0, 2, 0, 1}, {3, 2, 2}, in TEST()
Dlsh_projection_test.cc59 void SetInput(std::initializer_list<int> data) { in SetInput() function in tflite::__anone90781880111::LSHProjectionOpModel
83 m.SetInput({12345, 54321, 67890, 9876, -12345678}); in TEST()
95 m.SetInput({12345, 54321, 67890, 9876, -12345678}); in TEST()
106 m.SetInput({1234, 2345, 3456, 1234, 4567, 5678, 6789, 4567, 7891, 8912, in TEST()
Dbatch_to_space_nd_test.cc29 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anonb3939b050111::BatchToSpaceNDOpModel
99 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST()
108 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST()
129 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST()
Ddepthwise_conv_test.cc86 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anonec477e470111::DepthwiseConvolutionOpModel
98 m.SetInput({ in TEST()
124 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anonec477e470111::QuantizedDepthwiseConvolutionOpModel
151 m.SetInput({ in TEST()
Dbasic_rnn_test.cc154 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon221f2f940111::RNNOpModel
158 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon221f2f940111::RNNOpModel
244 rnn.SetInput(0, batch_start, batch_end); in TEST()
245 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
Dbidirectional_sequence_rnn_test.cc714 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon4fcb5fe30111::BidirectionalRNNOpModel
718 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon4fcb5fe30111::BidirectionalRNNOpModel
781 rnn.SetInput(0, batch_start, batch_end); in TEST()
782 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
824 rnn.SetInput(reverse_idx * rnn.input_size(), batch_start, batch_end); in TEST()
825 rnn.SetInput((rnn.sequence_len() + reverse_idx) * rnn.input_size(), in TEST()
891 rnn.SetInput(0, batch_start, batch_end); in TEST()
/external/tensorflow/tensorflow/contrib/lite/testing/
Dtf_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()
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()

12