mirror of https://github.com/alibaba/MNN.git
47 lines
2.1 KiB
C++
47 lines
2.1 KiB
C++
|
//
|
||
|
|
// FunctionSummary.hpp
|
||
|
|
// MNN
|
||
|
|
//
|
||
|
|
// Created by MNN on 2018/11/18.
|
||
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
||
|
|
//
|
||
|
|
|
||
|
|
// ========= CommonOptFunction.cpp ===========
|
||
|
|
|
||
|
|
void _SSE_MNNAddBias(float* dst, const float* bias, size_t planeNumber, size_t biasNumber);
|
||
|
|
|
||
|
|
void _SSE_MNNAddBiasRelu(float* dst, const float* bias, size_t planeNumber, size_t biasNumber);
|
||
|
|
|
||
|
|
void _SSE_MNNAddBiasRelu6(float* dst, const float* bias, size_t planeNumber, size_t biasNumber);
|
||
|
|
|
||
|
|
void _SSE_MNNCopyC4WithStride(const float* source, float* dest, size_t srcStride, size_t dstStride, size_t count);
|
||
|
|
|
||
|
|
void _SSE_MNNAddC4WithStride(const float* source, float* dest, size_t srcStride, size_t dstStride, size_t count);
|
||
|
|
|
||
|
|
// ========= MNNConvSlideWindowBorder.cpp ===========
|
||
|
|
|
||
|
|
void _SSE_MNNConvSlideWindowBorder(float* dst, const float* src, const float* weight, size_t src_depth_quad,
|
||
|
|
size_t src_depth_step, size_t fw, size_t fh, size_t weight_y_step, size_t weight_z_step,
|
||
|
|
size_t dilateX_step, size_t dilateY_step, float* alpha);
|
||
|
|
|
||
|
|
// ========= MNNConvSlideWindowMiddle.cpp ===========
|
||
|
|
|
||
|
|
void _SSE_MNNConvSlideWindowMiddle(float* dst, const float* src, const float* weight, size_t width, size_t src_w_setup,
|
||
|
|
size_t src_depth_quad, size_t src_depth_step, size_t fw, size_t fh, size_t dilateX_step,
|
||
|
|
size_t dilateY_step, float* alpha);
|
||
|
|
|
||
|
|
// ========= MNNGemmFloatCommon_4.cpp ===========
|
||
|
|
|
||
|
|
void _SSE_MNNGemmFloatCommon_4(float* dst, const float* src, const float* weight, size_t src_depth_quad, size_t dst_step,
|
||
|
|
size_t dst_depth_quad, size_t width, size_t weight_depth_offset);
|
||
|
|
|
||
|
|
// ========= MNNMatrixAdd.cpp ===========
|
||
|
|
|
||
|
|
void _SSE_MNNMatrixAdd(float* C, const float* A, const float* B, size_t widthC4, size_t cStride, size_t aStride,
|
||
|
|
size_t bStride, size_t height);
|
||
|
|
|
||
|
|
// ========= MNNMatrixSub.cpp ===========
|
||
|
|
|
||
|
|
void _SSE_MNNMatrixSub(float* C, const float* A, const float* B, size_t widthC4, size_t cStride, size_t aStride,
|
||
|
|
size_t bStride, size_t height);
|