/external/tensorflow/tensorflow/lite/kernels/ |
D | pack_test.cc | 48 void SetInput(int index, std::initializer_list<T> data) { in SetInput() function in tflite::__anoncf9bf68c0111::PackOpModel 62 model.SetInput(0, {1, 4}); in TEST() 63 model.SetInput(1, {2, 5}); in TEST() 64 model.SetInput(2, {3, 6}); in TEST() 72 model.SetInput(0, {1, 4}); in TEST() 73 model.SetInput(1, {2, 5}); in TEST() 74 model.SetInput(2, {3, 6}); in TEST() 82 model.SetInput(0, {1, 4}); in TEST() 83 model.SetInput(1, {2, 5}); in TEST() 84 model.SetInput(2, {3, 6}); in TEST() [all …]
|
D | concatenation_test.cc | 68 void SetInput(int index, std::initializer_list<float> data) { in SetInput() function in tflite::__anon7c4549130111::ConcatenationOpModel 79 void SetInput(int index, std::initializer_list<float> data) { in SetInput() function in tflite::__anon7c4549130111::QuantizedConcatenationOpModel 96 void SetInput(int index, std::initializer_list<bool> data) { in SetInput() function in tflite::__anon7c4549130111::BoolConcatenationOpModel 105 m0.SetInput(0, {1.0f, 3.0f, 4.0f, 7.0f}); in TEST() 113 m0.SetInput(0, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, in TEST() 123 m0.SetInput(0, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, in TEST() 125 m0.SetInput(1, {13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f, 19.0f, 20.0f, 21.0f, in TEST() 137 m0.SetInput(0, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, in TEST() 139 m0.SetInput(1, {13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f, 19.0f, 20.0f, 21.0f, in TEST() 153 m0.SetInput<uint8_t>(0, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, in TEST() [all …]
|
D | pooling_test.cc | 58 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon2e8345650111::FloatPoolingOpModel 69 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon2e8345650111::QuantizedPoolingOpModel 73 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon2e8345650111::QuantizedPoolingOpModel 88 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon2e8345650111::SymmetricQuantizedPoolingOpModel 92 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon2e8345650111::SymmetricQuantizedPoolingOpModel 107 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon2e8345650111::SymmetricQuantizedPoolingOpModel16 111 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon2e8345650111::SymmetricQuantizedPoolingOpModel16 145 m.SetInput({ in TEST() 159 m.SetInput({ in TEST() 173 m.SetInput({ in TEST() [all …]
|
D | gather_nd_test.cc | 45 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon40db79d90111::GatherNdOpModel 69 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST() 78 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST() 87 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST() 94 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST() 104 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST() 114 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST() 123 m.SetInput<float>({1.1, 1.2, 2.1, 2.2}); in TEST() 133 m.SetInput<float>({1.1, -1.2, 1.3, -2.1, 2.2, 2.3, // in TEST() 145 m.SetInput<float>({1.1, -1.2, 1.3, -2.1, 2.2, 2.3, // in TEST() [all …]
|
D | resize_bilinear_test.cc | 67 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anonb33b3d040111::ResizeBilinearOpModel 87 m.SetInput<float>({3, 6}); in TEST_P() 95 m.SetInput<uint8>({3, 6}); in TEST_P() 103 m.SetInput<int8_t>({3, 6}); in TEST_P() 114 m.SetInput<int16_t>({3, 6}); in TEST_P() 123 m.SetInput<float>({3, 9}); in TEST_P() 131 m.SetInput<uint8>({3, 9}); in TEST_P() 139 m.SetInput<int8_t>({3, 9}); in TEST_P() 150 m.SetInput<int16_t>({3, 9}); in TEST_P() 159 m.SetInput<float>({ in TEST_P() [all …]
|
D | resize_nearest_neighbor_test.cc | 67 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anona4c8837d0111::ResizeNearestNeighborOpModel 88 m.SetInput<float>({3, 6}); in TEST_P() 96 m.SetInput<uint8>({3, 6}); in TEST_P() 104 m.SetInput<int8_t>({-3, 6}); in TEST_P() 112 m.SetInput<int16_t>({-3, 6}); in TEST_P() 120 m.SetInput<float>({3, 9}); in TEST_P() 128 m.SetInput<uint8>({3, 9}); in TEST_P() 136 m.SetInput<int8_t>({3, -9}); in TEST_P() 144 m.SetInput<int16_t>({3, -9}); in TEST_P() 152 m.SetInput<float>({ in TEST_P() [all …]
|
D | activations_test.cc | 192 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon47e294320111::FloatActivationsOpModel 222 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon47e294320111::QuantizedActivationsOpModel 290 m.SetInput({ in TEST() 309 model.SetInput<int8_t>({ in TEST() 327 m.SetInput({ in TEST() 341 m.SetInput({ in TEST() 355 m.SetInput({ in TEST() 369 m.SetInput({ in TEST() 423 m.SetInput(float_input_values); in TestFloatHardSwish() 447 m.SetInput<QuantizedType>(float_input_values); in TestQuantizedHardSwish() [all …]
|
D | strided_slice_test.cc | 53 void SetInput(std::initializer_list<input_type> data) { in SetInput() function in tflite::__anon8b61d7eb0111::StridedSliceOpModel 56 void SetInput(const std::vector<input_type> data) { in SetInput() function in tflite::__anon8b61d7eb0111::StridedSliceOpModel 113 m.SetInput({1, 2, 3, 4}); in TYPED_TEST() 128 m.SetInput(values); in TYPED_TEST() 139 m.SetInput({1, 2, 3, 4}); in TYPED_TEST() 149 m.SetInput({1, 2, 3, 4}); in TYPED_TEST() 160 m.SetInput({1, 2, 3, 4}); in TYPED_TEST() 171 m.SetInput({1, 2, 3, 4}); in TYPED_TEST() 182 m.SetInput({1, 2, 3, 4}); in TYPED_TEST() 193 m.SetInput({1, 2, 3, 4}); in TYPED_TEST() [all …]
|
D | transpose_test.cc | 212 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anon9e70114c0111::TransposeOpModel 284 m.SetInput({1, 2, 3}); in TEST() 292 m.SetInput({1, 2, 3}); in TEST() 301 m.SetInput({0, 1, 2, 3, 4, 5}); in TEST() 309 m.SetInput({0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST() 320 m.SetInput({0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST() 331 m.SetInput({0, 1, 2, 3, 4, 5}); in TEST() 340 m.SetInput({0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST() 351 m.SetInput({0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST() 363 m.SetInput({0}); in TEST() [all …]
|
D | skip_gram_test.cc | 47 void SetInput(const string& content) { in SetInput() function in tflite::__anon461e746a0111::SkipGramOp 71 m.SetInput(kSentence); in TEST() 80 m.SetInput(kSentence); in TEST() 90 m.SetInput(kSentence); in TEST() 104 m.SetInput(kSentence); in TEST() 122 m.SetInput(kSentence); in TEST() 134 m.SetInput(kSentence); in TEST() 148 m.SetInput(kSentence); in TEST() 164 m.SetInput(kSentence); in TEST() 187 m.SetInput(kSentence); in TEST() [all …]
|
D | gather_test.cc | 46 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon95e652480111::GatherOpModel 77 m.SetInput<float>({-2.0, 0.2, 0.7, 0.8}); in TEST() 86 m.SetInput<float>({-2.0, 0.2, 0.7, 0.8}); in TEST() 98 m.SetInput<float>({1.0, 2.0, 3.0}); in TEST() 107 m.SetInput<float>({1.0, 3.0, 5.0}); in TEST() 116 m.SetInput<float>({1.0, 2.0, 3.0}); in TEST() 126 m.SetInput<float>({-2.0, 0.2, 0.7, 0.8}); in TEST() 136 m.SetInput<float>({-2.0, 0.2, 0.7, 0.8}); in TEST() 147 m.SetInput<float>({1, 2, 3, 4, 5, 6}); in TEST() 159 m.SetInput<float>({1, 2, 3, 4, 5, 6}); in TEST() [all …]
|
D | broadcast_to_test.cc | 57 void SetInput(std::initializer_list<InputType> data) { in SetInput() function in tflite::__anonbf80f4020111::BroadcastToOpModel 119 m.SetInput({3}); in TYPED_TEST() 127 m.SetInput({3, 4}); in TYPED_TEST() 136 m.SetInput({3, 4}); in TYPED_TEST() 144 m.SetInput({3}); in TYPED_TEST() 153 m.SetInput({3, 4}); in TYPED_TEST() 162 m.SetInput({3, 4}); in TYPED_TEST() 171 m.SetInput({1, 2, 3, 4, 5, 6}); in TYPED_TEST() 182 m.SetInput({1, 2, 3, 4, 5, 6}); in TYPED_TEST() 194 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TYPED_TEST() [all …]
|
D | conv_test.cc | 142 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anoncc1f39a30111::ConvolutionOpModel 170 m.SetInput({ in TEST_P() 201 m.SetInput({ in TEST_P() 236 m.SetInput({ in TEST_P() 266 m.SetInput({ in TEST_P() 293 m.SetInput({ in TEST_P() 321 m.SetInput({ in TEST_P() 355 m.SetInput({ in TEST_P() 389 m.SetInput({ in TEST_P() 425 m.SetInput({ in TEST_P() [all …]
|
D | slice_test.cc | 68 void SetInput(std::initializer_list<input_type> data) { in SetInput() function in tflite::__anon871d8d3d0111::SliceOpModel 96 m.SetInput({1, 2, 3, 4}); in TEST_P() 106 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST_P() 116 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST_P() 127 m.SetInput({1, 2, 3, 4, 5}); in TEST_P() 137 m.SetInput({1, 2, 3, 4}); in TEST_P() 147 m.SetInput({1, 2, 3, 4}); in TEST_P() 159 m.SetInput({1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6}); in TEST_P() 169 m.SetInput({1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6}); in TEST_P() 179 m.SetInput({1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6}); in TEST_P() [all …]
|
D | depthwise_conv_test.cc | 141 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anonfca7c40f0111::DepthwiseConvolutionOpModel 172 m.SetInput({ in TEST_P() 203 m.SetInput({ in TEST_P() 234 m.SetInput({ in TEST_P() 265 m.SetInput({ in StrideTest() 303 m.SetInput({ in PaddingTest() 338 m.SetInput({ in SimpleTest() 394 m.SetInput({0, 0, 0, 0, 0, 0, 0, 0, 0, in SimpleDilatedTestPaddingValid() 449 m.SetInput({1, 1, 1, 1, 1, 1, 1, 1, 1}); in SimpleDilatedTestPaddingSame() 491 m.SetInput({ in BatchPaddingValidTest() [all …]
|
D | batch_to_space_nd_test.cc | 35 void SetInput(std::initializer_list<T> data) { in SetInput() function in tflite::__anon8e376bcd0111::BatchToSpaceNDOpModel 114 m.SetInput<float>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST() 125 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST() 135 m.SetInput<float>({1, 2, 3, 4}); in TEST() 149 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST() 157 m.SetInput<float>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST() 169 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST() 181 m.SetInput<float>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST() 194 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST() 216 m.SetInput<float>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST() [all …]
|
D | embedding_lookup_test.cc | 52 void SetInput(std::initializer_list<int> data) { in SetInput() function in tflite::__anon71f285c50111::BaseEmbeddingLookupOpModel 108 m.SetInput({1, 0, 2}); in TEST() 124 m.SetInput({1, 0, 2}); in TEST() 145 m.SetInput({1, 0, 2}); in TEST() 166 m.SetInput({1, 0, 2}); in TEST() 187 m.SetInput({1, 0, 2}); in TEST() 208 m.SetInput({1, 0, 2}); in TEST() 229 m.SetInput({1, 0, 2}); in TEST() 250 m.SetInput({1, 0, 2}); in TEST()
|
D | fully_connected_test.cc | 268 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon1e118d2b0111::FloatFullyConnectedOpModel 271 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anon1e118d2b0111::FloatFullyConnectedOpModel 337 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon1e118d2b0111::QuantizedFullyConnectedOpModel 384 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::__anon1e118d2b0111::PerChannelQuantizedFullyConnectedOpModel 447 void SetInput(const std::vector<float>& f) { PopulateTensor(input_, f); } in SetInput() function in tflite::__anon1e118d2b0111::HybridFullyConnectedOpModel 518 m.SetInput({ in TEST_P() 537 m.SetInput({ in TEST_P() 603 m.SetInput({ in TEST() 653 m.SetInput<uint8_t>({ in TEST_P() 687 m.SetInput<uint8_t>({ in TEST_P() [all …]
|
D | unidirectional_sequence_rnn_test.cc | 219 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anona0a7fc2c0111::UnidirectionalRNNOpModel 223 void SetInput(int offset, float* begin, float* end) { in SetInput() function in tflite::__anona0a7fc2c0111::UnidirectionalRNNOpModel 289 rnn.SetInput(0, batch_start, batch_end); in TEST() 290 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST() 318 rnn.SetInput(0, batch_start, batch_end); in TEST_P() 319 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST_P() 345 rnn.SetInput(0, batch_start, batch_end); in TEST_P() 346 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST_P() 372 rnn.SetInput(2 * i * rnn.input_size(), batch_start, batch_end); in TEST() 373 rnn.SetInput((2 * i + 1) * rnn.input_size(), batch_start, batch_end); in TEST() [all …]
|
D | pooling3d_test.cc | 63 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::BasePoolingOpModel 113 void BasePoolingOpModel<float>::SetInput(const std::vector<float>& data) { in SetInput() function in tflite::BasePoolingOpModel 165 m.SetInput({0, 6, 2, 4, 4, 5, 1, 4, 3, 2, 10, 7, 2, 3, 5, 1}); in TYPED_TEST() 177 m.SetInput({0, 6, 2, 4, 4, 5, 1, 4, 3, 2, 10, 7, 2, 3, 5, 1}); in TYPED_TEST() 192 m.SetInput({0, 6, 2, 4, 2, 5, 4, 3, 3, 2, 10, 7, 3, 2, 2, 4}); in TYPED_TEST() 209 m.SetInput({0, 6, 2, 4, 2, 5, 4, 3, 3, 2, 10, 7, 3, 2, 2, 4}); in TYPED_TEST() 221 m.SetInput({0, 6, 2, 4, 4, 5, 1, 4, 3, 2, 10, 7, 2, 3, 5, 1}); in TYPED_TEST() 233 m.SetInput({0, 6, 2, 4, 4, 5, 1, 4, 3, 2, 10, 7, 2, 3, 5, 1}); in TYPED_TEST() 248 m.SetInput({0, 6, 2, 4, 2, 5, 4, 3, 3, 2, 10, 7, 3, 2, 2, 4}); in TYPED_TEST() 264 m.SetInput({0, 6, 2, 4, 2, 5, 4, 3, 3, 2, 10, 7, 3, 2, 2, 4}); in TYPED_TEST()
|
D | transpose_conv_test.cc | 96 void SetInput(std::initializer_list<float> data) { in SetInput() function in tflite::__anona9e836210111::BaseTransposeConvOpModel 150 model.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST_P() 178 model.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST_P() 207 model.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST_P() 236 model.SetInput({1, 2, 3, 4}); in TEST_P() 262 model.SetInput({1, 2, 3, 4}); in TEST_P() 291 model.SetInput({323, 521}); in TEST_P() 320 model.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); in TEST_P() 344 model.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST_P() 368 model.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, in TEST_P() [all …]
|
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
D | concat_test.cc | 66 void SetInput(int index, std::vector<float> data) { in SetInput() function in tflite::QuantizedConcatenationOpModel 93 m0.SetInput<integer_type>(0, {1.0f, 3.0f, 4.0f, 7.0f}); in FourInputsQuantizedSameRangeImpl() 94 m0.SetInput<integer_type>(1, {1.1f, 3.1f, 4.1f, 7.1f}); in FourInputsQuantizedSameRangeImpl() 95 m0.SetInput<integer_type>(2, {1.2f, 3.2f, 4.2f, 7.2f}); in FourInputsQuantizedSameRangeImpl() 96 m0.SetInput<integer_type>(3, {1.3f, 3.3f, 4.3f, 7.3f}); in FourInputsQuantizedSameRangeImpl() 133 m0.SetInput<integer_type>(0, tensor0); in TwoInputsNegativeAxisImpl() 134 m0.SetInput<integer_type>(1, tensor1); in TwoInputsNegativeAxisImpl() 158 m0.SetInput<uint8_t>(0, {1.0f, 3.0f, 4.0f, 7.0f}); in TEST() 159 m0.SetInput<uint8_t>(1, {1.1f, 3.1f, 4.1f, 7.1f}); in TEST() 160 m0.SetInput<uint8_t>(2, {1.2f, 3.2f, 4.2f, 7.2f}); in TEST() [all …]
|
D | quantize_test.cc | 32 void SetInput(const std::vector<float>& data) { in SetInput() function in tflite::QuantizeOpModel 55 m.SetInput<uint8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST() 69 m.SetInput<uint8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST() 83 m.SetInput<uint8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST() 97 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST() 111 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST() 125 m.SetInput<uint8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST() 138 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST() 151 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST() 164 m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10}); in TEST()
|
D | pack_test.cc | 41 void SetInput(int index, std::initializer_list<float> data) { in SetInput() function in tflite::PackOpModel 70 model.SetInput<typename TestFixture::TypeToTest>(0, {1, 4}); in TYPED_TEST() 71 model.SetInput<typename TestFixture::TypeToTest>(1, {2, 5}); in TYPED_TEST() 72 model.SetInput<typename TestFixture::TypeToTest>(2, {3, 6}); in TYPED_TEST() 85 model.SetInput<typename TestFixture::TypeToTest>(0, {1, 4}); in TYPED_TEST() 86 model.SetInput<typename TestFixture::TypeToTest>(1, {2, 5}); in TYPED_TEST() 87 model.SetInput<typename TestFixture::TypeToTest>(2, {3, 6}); in TYPED_TEST() 100 model.SetInput<typename TestFixture::TypeToTest>(0, {1, 4}); in TYPED_TEST() 101 model.SetInput<typename TestFixture::TypeToTest>(1, {2, 5}); in TYPED_TEST() 102 model.SetInput<typename TestFixture::TypeToTest>(2, {3, 6}); in TYPED_TEST() [all …]
|
D | strided_slice_test.cc | 49 void SetInput(std::initializer_list<input_type> data) { in SetInput() function in tflite::StridedSliceOpModel 77 m.SetInput({1, 2, 3, 4}); in TEST() 94 m.SetInput({1, 2, 3, 4}); in TEST() 111 m.SetInput({1, 2, 3, 4}); in TEST() 128 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST() 145 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST() 162 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST() 179 m.SetInput({0, 1, 2, 3}); in TEST() 196 m.SetInput({1, 2, 3, 4, 5, 6}); in TEST() 213 m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TEST() [all …]
|