PALSelectV2.h

Go to the documentation of this file.

/*
 * Copyright (c) 2023 Samsung Electronics Co., Ltd. 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 LUCI_INTERPRETER_PAL_SELECT_V2_COMMON_H
#define LUCI_INTERPRETER_PAL_SELECT_V2_COMMON_H
 
#include "PALUtils.h"
#include "ProcessBroadcastShapes.h"
 
namespace luci_interpreter_pal
{
 
template <typename D, typename T>
void Select(const luci_interpreter::RuntimeShape &input_condition_shape,
            const D *input_condition_data, const luci_interpreter::RuntimeShape &input_x_shape,
            const T *input_x_data, const luci_interpreter::RuntimeShape &input_y_shape,
            const T *input_y_data, const luci_interpreter::RuntimeShape &output_shape,
            T *output_data)
{
  int64_t flatsize;
  // Allow select operator executions on mixed scalar tensors and one element
  // tensors.
  if (input_condition_shape.flatSize() == 1 && input_x_shape.flatSize() == 1 &&
      input_y_shape.flatSize() == 1 && output_shape.flatSize() == 1)
  {
    flatsize = 1;
  }
  else
  {
    flatsize = input_condition_shape.flatSize();
  }
  for (int64_t i = 0; i < flatsize; ++i)
  {
    output_data[i] = input_condition_data[i] ? input_x_data[i] : input_y_data[i];
  }
}
void Select(const luci_interpreter::RuntimeShape &input_condition_shape, {…}
 
} // namespace luci_interpreter_pal
 
#endif // LUCI_INTERPRETER_PAL_SELECT_V2_COMMON_H