diff options
author | Yu-Cheng Ling <ycling@google.com> | 2018-05-13 19:52:18 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-13 19:55:02 -0700 |
commit | 699b217cd6c5ddc0832be8471dde47999829e435 (patch) | |
tree | 035167be1ec270dded665347d20ec9385bed0fcc /tensorflow/contrib/lite/kernels/register.cc | |
parent | 2fbc0c5a45955c877e0a165bb561fc2f01518321 (diff) |
Introduce op version into TFLite
PiperOrigin-RevId: 196448769
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/register.cc | 23 |
1 files changed, 0 insertions, 23 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc index d7eed96db0..0c7cfcaf10 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -167,29 +167,6 @@ BuiltinOpResolver::BuiltinOpResolver() { tflite::ops::custom::Register_AUDIO_SPECTROGRAM()); } -TfLiteRegistration* BuiltinOpResolver::FindOp( - tflite::BuiltinOperator op) const { - auto it = builtins_.find(op); - return it != builtins_.end() ? it->second : nullptr; -} - -TfLiteRegistration* BuiltinOpResolver::FindOp(const char* op) const { - auto it = custom_ops_.find(op); - return it != custom_ops_.end() ? it->second : nullptr; -} - -void BuiltinOpResolver::AddBuiltin(tflite::BuiltinOperator op, - TfLiteRegistration* registration) { - registration->builtin_code = op; - builtins_.insert(std::make_pair(op, registration)); -} - -void BuiltinOpResolver::AddCustom(const char* name, - TfLiteRegistration* registration) { - registration->builtin_code = BuiltinOperator_CUSTOM; - custom_ops_.insert(std::make_pair(std::string(name), registration)); -} - } // namespace builtin } // namespace ops } // namespace tflite |