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

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

int nnfw::cker::Clamp (const int v, const int lo, const int hi)
 
void nnfw::cker::StridedSlicePadIndices (StridedSliceParams *p, int dim_count)
 
int nnfw::cker::StartForAxis (const StridedSliceParams &params, const Shape &input_shape, int axis)
 
int nnfw::cker::StopForAxis (const StridedSliceParams &params, const Shape &input_shape, int axis, int start_for_axis)
 
bool nnfw::cker::LoopCondition (int index, int stop, int stride)
 
template<typename T >
StridedSliceParams nnfw::cker::buildStridedSliceParams (const T *begin, const T *end, const T *strides, const uint32_t begin_mask, const uint32_t end_mask, const uint32_t shrink_axis_mask, const uint8_t rank)
 
void nnfw::cker::checkOutputSize (const StridedSliceParams &op_params, const Shape &input_shape, const Shape &output_shape, uint32_t rank)
 
template<typename T >
void nnfw::cker::StridedSlice (const StridedSliceParams &op_params, const Shape &unextended_input_shape, const T *input_data, const Shape &unextended_output_shape, T *output_data)