Home
last modified time | relevance | path

Searched refs:GenericPrepare (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Delementwise.cc40 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare() function
120 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_ABS()
132 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_SIN()
144 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_COS()
156 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_LOG()
168 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_SQRT()
180 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_RSQRT()
192 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_SQUARE()
204 elementwise::GenericPrepare<elementwise::IsLogicalSupportedType>, in Register_LOGICAL_NOT()
Delu.cc87 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare() function
112 return GenericPrepare(context, node); in EluPrepare()
/external/tensorflow/tensorflow/lite/kernels/
Delementwise.cc84 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare() function
302 elementwise::GenericPrepare<elementwise::IsAbsSupportedType, in Register_ABS()
311 elementwise::GenericPrepare<elementwise::IsNumericSupportedType, in Register_SIN()
320 elementwise::GenericPrepare<elementwise::IsNumericSupportedType, in Register_COS()
329 elementwise::GenericPrepare<elementwise::IsNumericSupportedType, in Register_LOG()
338 elementwise::GenericPrepare<elementwise::IsNumericSupportedType, in Register_SQRT()
348 elementwise::GenericPrepare<elementwise::IsRsqrtSupportedType, in Register_RSQRT()
357 elementwise::GenericPrepare<elementwise::IsNumericSupportedType, in Register_SQUARE()
366 elementwise::GenericPrepare<elementwise::IsLogicalSupportedType, in Register_LOGICAL_NOT()
Dpooling.cc68 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare() function
462 pooling::GenericPrepare<pooling::kAverage>, in Register_AVERAGE_POOL_REF()
469 pooling::GenericPrepare<pooling::kMax>, in Register_MAX_POOL_REF()
476 pooling::GenericPrepare<pooling::kL2>, in Register_L2_POOL_REF()
483 pooling::Init, pooling::Free, pooling::GenericPrepare<pooling::kAverage>, in Register_AVERAGE_POOL_GENERIC_OPT()
490 pooling::GenericPrepare<pooling::kMax>, in Register_MAX_POOL_GENERIC_OPT()
497 pooling::GenericPrepare<pooling::kL2>, in Register_L2_POOL_GENERIC_OPT()
Dactivations.cc252 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare() function
312 TF_LITE_ENSURE_STATUS(GenericPrepare(context, node)); in HardSwishPrepare()
1421 return GenericPrepare(context, node); in EluPrepare()