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

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::ruy
 

Functions

template<typename T >
void nnfw::ruy::ExtractPatchIntoBufferColumn (const Shape &input_shape, int w, int h, int b, int kheight, int kwidth, int stride_width, int stride_height, int pad_width, int pad_height, int in_width, int in_height, int in_depth, int single_buffer_length, int buffer_id, const T *in_data, T *conv_buffer_data, uint8_t zero_byte)
 
template<typename T >
void nnfw::ruy::DilatedIm2col (const ConvParams &params, const Shape &input_shape, const T *input_data, const Shape &filter_shape, const Shape &output_shape, T *im2col_data, const int32_t *zero_bytes, const int zero_bytes_len)
 
template<typename T >
void nnfw::ruy::DilatedIm2col (const ConvParams &params, uint8_t zero_byte, const Shape &input_shape, const T *input_data, const Shape &filter_shape, const Shape &output_shape, T *im2col_data)
 
template<typename T >
void nnfw::ruy::Im2col (const ConvParams &params, int kheight, int kwidth, uint8_t zero_byte, const Shape &input_shape, const T *input_data, const Shape &output_shape, T *output_data)