ONE - On-device Neural Engine
Loading...
Searching...
No Matches
RoPE.h File Reference
#include <stdexcept>
#include "cker/Shape.h"
#include "cker/Types.h"
#include "cker/Utils.h"

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

template<typename T >
void nnfw::cker::RoPE (const RoPEMode mode, const Shape &input_shape, const T *input_data, const Shape &sin_table_shape, const T *sin_table_data, const Shape &cos_table_shape, const T *cos_table_data, const Shape &output_shape, T *output_data)