floor.cc

00001 /* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
00002 
00003 Licensed under the Apache License, Version 2.0 (the "License");
00004 you may not use this file except in compliance with the License.
00005 You may obtain a copy of the License at
00006 
00007     http://www.apache.org/licenses/LICENSE-2.0
00008 
00009 Unless required by applicable law or agreed to in writing, software
00010 distributed under the License is distributed on an "AS IS" BASIS,
00011 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
00012 See the License for the specific language governing permissions and
00013 limitations under the License.
00014 ==============================================================================*/
00015 
00016 #include "tensorflow/lite/c/c_api_internal.h"
00017 #include "tensorflow/lite/kernels/internal/reference/floor.h"
00018 #include "tensorflow/lite/kernels/internal/tensor_ctypes.h"
00019 #include "tensorflow/lite/kernels/kernel_util.h"
00020 
00021 namespace tflite {
00022 namespace ops {
00023 namespace micro {
00024 namespace floor {
00025 
00026 constexpr int kInputTensor = 0;
00027 constexpr int kOutputTensor = 0;
00028 
00029 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
00030   const TfLiteTensor* input = GetInput(context, node, kInputTensor);
00031   TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32);
00032   TfLiteTensor* output = GetOutput(context, node, kOutputTensor);
00033   reference_ops::Floor(GetTensorShape(input), GetTensorData<float>(input),
00034                        GetTensorShape(output), GetTensorData<float>(output));
00035   return kTfLiteOk;
00036 }
00037 }  // namespace floor
00038 
00039 TfLiteRegistration* Register_FLOOR() {
00040   static TfLiteRegistration r = {/*init=*/nullptr,
00041                                  /*free=*/nullptr, /*prepare=*/nullptr,
00042                                  floor::Eval};
00043   return &r;
00044 }
00045 
00046 }  // namespace micro
00047 }  // namespace ops
00048 }  // namespace tflite