ac8f6785d3
This kernel currently only supports float type and has filter with the same data format as TensorFlow Conv3D op. The conversion change will be added in a follow-up cl. PiperOrigin-RevId: 351726918 Change-Id: Id2cc805cb89a1da1fda656a2260516eda8bd9119
166 lines
6.7 KiB
C++
166 lines
6.7 KiB
C++
/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
==============================================================================*/
|
|
#ifndef TENSORFLOW_LITE_KERNELS_BUILTIN_OP_KERNELS_H_
|
|
#define TENSORFLOW_LITE_KERNELS_BUILTIN_OP_KERNELS_H_
|
|
|
|
#include "tensorflow/lite/c/common.h"
|
|
|
|
namespace tflite {
|
|
namespace ops {
|
|
namespace builtin {
|
|
|
|
// Forward declaration of all builtin op kernel registration methods. These
|
|
// registrations are included with the standard `BuiltinOpResolver`.
|
|
//
|
|
// This header is particularly useful in cases where only a subset of ops are
|
|
// needed. In such cases, the client can selectively add only the registrations
|
|
// their model requires, using a custom `OpResolver` or `MutableOpResolver`.
|
|
// Selective registration in turn allows the linker to strip unused kernels.
|
|
|
|
TfLiteRegistration* Register_ABS();
|
|
TfLiteRegistration* Register_ADD();
|
|
TfLiteRegistration* Register_ADD_N();
|
|
TfLiteRegistration* Register_ARG_MAX();
|
|
TfLiteRegistration* Register_ARG_MIN();
|
|
TfLiteRegistration* Register_AVERAGE_POOL_2D();
|
|
TfLiteRegistration* Register_BATCH_TO_SPACE_ND();
|
|
TfLiteRegistration* Register_BATCH_MATMUL();
|
|
TfLiteRegistration* Register_BIDIRECTIONAL_SEQUENCE_LSTM();
|
|
TfLiteRegistration* Register_BIDIRECTIONAL_SEQUENCE_RNN();
|
|
TfLiteRegistration* Register_BROADCAST_TO();
|
|
TfLiteRegistration* Register_CALL_ONCE();
|
|
TfLiteRegistration* Register_CAST();
|
|
TfLiteRegistration* Register_CEIL();
|
|
TfLiteRegistration* Register_CONCATENATION();
|
|
TfLiteRegistration* Register_CONV_2D();
|
|
TfLiteRegistration* Register_CONV_3D();
|
|
TfLiteRegistration* Register_COS();
|
|
TfLiteRegistration* Register_CUMSUM();
|
|
TfLiteRegistration* Register_DENSIFY();
|
|
TfLiteRegistration* Register_DEPTH_TO_SPACE();
|
|
TfLiteRegistration* Register_DEPTHWISE_CONV_2D();
|
|
TfLiteRegistration* Register_DEQUANTIZE();
|
|
TfLiteRegistration* Register_DIV();
|
|
TfLiteRegistration* Register_ELU();
|
|
TfLiteRegistration* Register_EMBEDDING_LOOKUP();
|
|
TfLiteRegistration* Register_EMBEDDING_LOOKUP_SPARSE();
|
|
TfLiteRegistration* Register_EQUAL();
|
|
TfLiteRegistration* Register_EXP();
|
|
TfLiteRegistration* Register_EXPAND_DIMS();
|
|
TfLiteRegistration* Register_FAKE_QUANT();
|
|
TfLiteRegistration* Register_FILL();
|
|
TfLiteRegistration* Register_FLOOR();
|
|
TfLiteRegistration* Register_FLOOR_DIV();
|
|
TfLiteRegistration* Register_FLOOR_MOD();
|
|
TfLiteRegistration* Register_FULLY_CONNECTED();
|
|
TfLiteRegistration* Register_GATHER();
|
|
TfLiteRegistration* Register_GATHER_ND();
|
|
TfLiteRegistration* Register_GREATER();
|
|
TfLiteRegistration* Register_GREATER_EQUAL();
|
|
TfLiteRegistration* Register_HARD_SWISH();
|
|
TfLiteRegistration* Register_HASHTABLE_LOOKUP();
|
|
TfLiteRegistration* Register_IF();
|
|
TfLiteRegistration* Register_L2_NORMALIZATION();
|
|
TfLiteRegistration* Register_L2_POOL_2D();
|
|
TfLiteRegistration* Register_LEAKY_RELU();
|
|
TfLiteRegistration* Register_LESS();
|
|
TfLiteRegistration* Register_LESS_EQUAL();
|
|
TfLiteRegistration* Register_LOCAL_RESPONSE_NORMALIZATION();
|
|
TfLiteRegistration* Register_LOG();
|
|
TfLiteRegistration* Register_LOGICAL_AND();
|
|
TfLiteRegistration* Register_LOGICAL_NOT();
|
|
TfLiteRegistration* Register_LOGICAL_OR();
|
|
TfLiteRegistration* Register_LOGISTIC();
|
|
TfLiteRegistration* Register_LOG_SOFTMAX();
|
|
TfLiteRegistration* Register_LSH_PROJECTION();
|
|
TfLiteRegistration* Register_LSTM();
|
|
TfLiteRegistration* Register_MATRIX_DIAG();
|
|
TfLiteRegistration* Register_MATRIX_SET_DIAG();
|
|
TfLiteRegistration* Register_MAXIMUM();
|
|
TfLiteRegistration* Register_MAX_POOL_2D();
|
|
TfLiteRegistration* Register_MEAN();
|
|
TfLiteRegistration* Register_MINIMUM();
|
|
TfLiteRegistration* Register_MIRROR_PAD();
|
|
TfLiteRegistration* Register_MUL();
|
|
TfLiteRegistration* Register_NEG();
|
|
TfLiteRegistration* Register_NON_MAX_SUPPRESSION_V4();
|
|
TfLiteRegistration* Register_NON_MAX_SUPPRESSION_V5();
|
|
TfLiteRegistration* Register_NOT_EQUAL();
|
|
TfLiteRegistration* Register_ONE_HOT();
|
|
TfLiteRegistration* Register_PACK();
|
|
TfLiteRegistration* Register_PAD();
|
|
TfLiteRegistration* Register_PADV2();
|
|
TfLiteRegistration* Register_POW();
|
|
TfLiteRegistration* Register_PRELU();
|
|
TfLiteRegistration* Register_QUANTIZE();
|
|
TfLiteRegistration* Register_RANGE();
|
|
TfLiteRegistration* Register_RANK();
|
|
TfLiteRegistration* Register_REDUCE_ANY();
|
|
TfLiteRegistration* Register_REDUCE_MAX();
|
|
TfLiteRegistration* Register_REDUCE_MIN();
|
|
TfLiteRegistration* Register_REDUCE_PROD();
|
|
TfLiteRegistration* Register_RELU();
|
|
TfLiteRegistration* Register_RELU6();
|
|
TfLiteRegistration* Register_RELU_N1_TO_1();
|
|
TfLiteRegistration* Register_RESHAPE();
|
|
TfLiteRegistration* Register_RESIZE_BILINEAR();
|
|
TfLiteRegistration* Register_RESIZE_NEAREST_NEIGHBOR();
|
|
TfLiteRegistration* Register_REVERSE_SEQUENCE();
|
|
TfLiteRegistration* Register_REVERSE_V2();
|
|
TfLiteRegistration* Register_RFFT2D();
|
|
TfLiteRegistration* Register_RNN();
|
|
TfLiteRegistration* Register_ROUND();
|
|
TfLiteRegistration* Register_RSQRT();
|
|
TfLiteRegistration* Register_SCATTER_ND();
|
|
TfLiteRegistration* Register_SEGMENT_SUM();
|
|
TfLiteRegistration* Register_SELECT();
|
|
TfLiteRegistration* Register_SELECT_V2();
|
|
TfLiteRegistration* Register_SHAPE();
|
|
TfLiteRegistration* Register_SIN();
|
|
TfLiteRegistration* Register_SKIP_GRAM();
|
|
TfLiteRegistration* Register_SLICE();
|
|
TfLiteRegistration* Register_SOFTMAX();
|
|
TfLiteRegistration* Register_SPACE_TO_BATCH_ND();
|
|
TfLiteRegistration* Register_SPACE_TO_DEPTH();
|
|
TfLiteRegistration* Register_SPARSE_TO_DENSE();
|
|
TfLiteRegistration* Register_SPLIT();
|
|
TfLiteRegistration* Register_SPLIT_V();
|
|
TfLiteRegistration* Register_SQRT();
|
|
TfLiteRegistration* Register_SQUARE();
|
|
TfLiteRegistration* Register_SQUARED_DIFFERENCE();
|
|
TfLiteRegistration* Register_SQUEEZE();
|
|
TfLiteRegistration* Register_STRIDED_SLICE();
|
|
TfLiteRegistration* Register_SUB();
|
|
TfLiteRegistration* Register_SUM();
|
|
TfLiteRegistration* Register_SVDF();
|
|
TfLiteRegistration* Register_TANH();
|
|
TfLiteRegistration* Register_TILE();
|
|
TfLiteRegistration* Register_TOPK_V2();
|
|
TfLiteRegistration* Register_TRANSPOSE();
|
|
TfLiteRegistration* Register_TRANSPOSE_CONV();
|
|
TfLiteRegistration* Register_UNIDIRECTIONAL_SEQUENCE_LSTM();
|
|
TfLiteRegistration* Register_UNIDIRECTIONAL_SEQUENCE_RNN();
|
|
TfLiteRegistration* Register_UNIQUE();
|
|
TfLiteRegistration* Register_UNPACK();
|
|
TfLiteRegistration* Register_WHERE();
|
|
TfLiteRegistration* Register_WHILE();
|
|
TfLiteRegistration* Register_ZEROS_LIKE();
|
|
|
|
} // namespace builtin
|
|
} // namespace ops
|
|
} // namespace tflite
|
|
|
|
#endif // TENSORFLOW_LITE_KERNELS_BUILTIN_OP_KERNELS_H_
|