Home
last modified time | relevance | path

Searched refs:GetRegistration (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv_test.cc164 GetRegistration(), {TensorType_FLOAT32, {1, 3, 2, 2}}, in TEST_P()
195 GetRegistration(), {TensorType_FLOAT32, {1, 3, 2, 2}}, in TEST_P()
226 GetRegistration(), {TensorType_FLOAT32, {1, 3, 2, 2}}, in TEST_P()
286 StrideTest(GetRegistration(), /*num_thread=*/1); in TEST_P()
290 StrideTest(GetRegistration(), /*num_thread=*/4); in TEST_P()
325 PaddingTest(GetRegistration(), /*num_thread=*/1); in TEST_P()
329 PaddingTest(GetRegistration(), /*num_thread=*/4); in TEST_P()
360 SimpleTest(GetRegistration(), /*num_thread=*/1); in TEST_P()
364 SimpleTest(GetRegistration(), /*num_thread=*/4); in TEST_P()
422 SimpleDilatedTestPaddingValid(GetRegistration(), /*num_thread=*/1); in TEST_P()
[all …]
Dconv_test.cc166 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST_P()
196 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST_P()
232 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 2}}, in TEST_P()
262 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST_P()
287 GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST_P()
320 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST_P()
354 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST_P()
390 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {1, 2, 4, 1}}, in TEST_P()
424 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 2}}, in TEST_P()
457 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 2}}, in TEST_P()
[all …]
Dtranspose_conv_test.cc132 TfLiteRegistration* GetRegistration() { in GetRegistration() function in tflite::__anon7fbb49030111::TransposeConvOpTest
147 GetRegistration(), {1, 4, 4, 1}, {TensorType_FLOAT32, {1, 3, 3, 1}}, in TEST_P()
174 GetRegistration(), {1, 4, 4, 1}, {TensorType_FLOAT32, {1, 3, 3, 2}}, in TEST_P()
203 GetRegistration(), {1, 6, 6, 1}, {TensorType_FLOAT32, {1, 3, 3, 2}}, in TEST_P()
233 GetRegistration(), {1, 5, 5, 1}, {TensorType_FLOAT32, {1, 3, 3, 1}}, in TEST_P()
258 GetRegistration(), {1, 5, 5, 2}, {TensorType_FLOAT32, {2, 3, 3, 1}}, in TEST_P()
288 GetRegistration(), {1, 3, 4, 1}, {TensorType_FLOAT32, {1, 3, 3, 1}}, in TEST_P()
316 GetRegistration(), {1, 4, 4, 1}, in TEST_P()
340 GetRegistration(), {1, 4, 4, 1}, in TEST_P()
364 GetRegistration(), {1, 6, 6, 1}, in TEST_P()
[all …]
Dfully_connected_test.cc401 FloatFullyConnectedOpModel m(GetRegistration(), /*units=*/3, /*batches=*/2, in TEST_P()
422 FloatFullyConnectedOpModel m(GetRegistration(), /*units=*/1, /*batches=*/2, in TEST_P()
465 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
495 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
529 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
556 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
583 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
614 GetRegistration(), /*units=*/4, /*batches*/ 1, in TEST_P()
642 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
673 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
[all …]
Ddepthwise_conv_hybrid_test.cc328 GetRegistration(), {TensorType_FLOAT32, {1, 2, 3, 2}}, in TEST_P()
374 GetRegistration(), {TensorType_FLOAT32, {1, 3, 3, 8}}, in TEST_P()
410 GetRegistration(), {TensorType_FLOAT32, {1, 3, 3, 8}}, in TEST_P()
Dactivations_test.cc500 FloatActivationsOpModel m(GetRegistration(), BuiltinOperator_TANH, in TEST_P()
763 GetRegistration(), BuiltinOperator_TANH, in TEST_P()
825 GetRegistration(), BuiltinOperator_TANH, in TEST_P()
886 GetRegistration(), BuiltinOperator_TANH, in TEST_P()
990 GetRegistration(), BuiltinOperator_TANH, in TEST_P()
1004 FloatActivationsOpModel m(GetRegistration(), BuiltinOperator_LOGISTIC, in TEST_P()
1018 QuantizedActivationsOpModel m(GetRegistration(), BuiltinOperator_LOGISTIC, in TEST_P()
1071 QuantizedActivationsOpModel m(GetRegistration(), BuiltinOperator_LOGISTIC, in TEST_P()
1127 GetRegistration(), BuiltinOperator_LOGISTIC, in TEST_P()
1223 GetRegistration(), BuiltinOperator_LOGISTIC, in TEST_P()
Dtest_util.h904 TfLiteRegistration* GetRegistration() { in GetRegistration() function