| 1234567891011121314151617181920212223242526272829303132333435 |
- /* Copyright 2020 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_INTERNAL_MAX_H_
- #define TENSORFLOW_LITE_KERNELS_INTERNAL_MAX_H_
- #include <cmath>
- namespace tflite {
- #if defined(TF_LITE_USE_GLOBAL_MAX) || defined(__ZEPHYR__)
- inline float TfLiteMax(const float& x, const float& y) {
- return ::max(x, y);
- }
- #else
- template <class T>
- inline T TfLiteMax(const T& x, const T& y) {
- return ::fmax(x, y);
- }
- #endif
- } // namespace tflite
- #endif // TENSORFLOW_LITE_KERNELS_INTERNAL_MAX_H_
|