Skip to content

Commit

Permalink
[onert-micro] fix build error with cmsisnn
Browse files Browse the repository at this point in the history
To build onert_micro_arm target with -DKERNELS=cmsisnn, we need this
header.

ONE-DCO-1.0-Signed-off-by: Chunseok Lee <[email protected]>
  • Loading branch information
chunseoklee authored and chunseoklee committed Aug 16, 2024
1 parent 06be246 commit 4d4bcce
Showing 1 changed file with 39 additions and 0 deletions.
39 changes: 39 additions & 0 deletions onert-micro/onert-micro/include/pal/cmsisnn/PALArgMax.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
/*
* Copyright (c) 2024 Samsung Electronics Co., Ltd. All Rights Reserved
* Copyright 2017 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 ONERT_MICRO_EXECUTE_PAL_ARG_MAX_H
#define ONERT_MICRO_EXECUTE_PAL_ARG_MAX_H

#include "PALArgMinMaxCommon.h"
namespace onert_micro
{
namespace execute
{
namespace pal
{

template <typename T1, typename T2, typename T3>
OMStatus ArgMax(const core::OMRuntimeShape &input1_shape, const T1 *input1_data,
const T3 *input2_data, const core::OMRuntimeShape &output_shape, T2 *output_data)
{
return ArgMinMax(input1_shape, input1_data, input2_data, output_shape, output_data,
std::greater<T1>());
}
} // namespace pal
} // namespace execute
} // namespace onert_micro
#endif // ONERT_MICRO_EXECUTE_PAL_ARG_MAX_H

0 comments on commit 4d4bcce

Please sign in to comment.