ONE - On-device Neural Engine
Loading...
Searching...
No Matches
PALGatherND.h File Reference
#include "PALUtils.h"
#include <cmath>

Go to the source code of this file.

Namespaces

namespace  luci_interpreter_pal
 

Functions

template<typename ParamsT , typename IndicesT >
void luci_interpreter_pal::GatherND (luci_interpreter::RuntimeShape params_shape, const ParamsT *param_data, luci_interpreter::RuntimeShape indices_shape, const IndicesT *index_data, ParamsT *output_data)
 

Variables

constexpr int luci_interpreter_pal::MAX_INDICES_ND = 5