前端和昇腾处理器之间的连接链路模块

/** * Copyright 2019-2020 Huawei Technologies Co., Ltd * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #ifndef DNN_OP_H__ #define DNN_OP_H__ #include "cce/blas_struct.h" #include "cce/cce.h" #include "cce/customize.h" namespace cce { /** * @ingroup dnn * @brief create descriptor of parameters for exponential function * @param [in] point to descriptor of parameters for exponential function * @return ccStatus_t */ ccStatus_t ccCreateExpDescriptor(ccExpDescriptor_t *expDesc); /** * @ingroup dnn * @brief create descriptor of parameters for logarithmic function * @param [in] point to descriptor of parameters for logarithmic function * @return ccStatus_t */ ccStatus_t ccCreateLogDescriptor(ccLogDescriptor_t *logDesc); /** * @ingroup dnn * @brief create descriptor of parameters for pow function * @param [in] point to descriptor of parameters for pow function * @return ccStatus_t */ ccStatus_t ccCreatePowDescriptor(ccPowDescriptor_t *powDesc); /** * @ingroup dnn * @brief destroy descriptor of parameters for exponential function * @param [in] point to descriptor of parameters for exponential function * @return ccStatus_t */ ccStatus_t ccDestroyExpDescriptor(ccExpDescriptor_t *expDesc); /** * @ingroup dnn * @brief destroy descriptor of parameters for logarithmic function * @param [in] point to descriptor of parameters for exponential function * @return ccStatus_t */ ccStatus_t ccDestroyLogDescriptor(ccLogDescriptor_t *logDesc); /** * @ingroup dnn * @brief destroy descriptor of parameters for pow function * @param [in] point to descriptor of parameters for pow function * @return ccStatus_t */ ccStatus_t ccDestroyPowDescriptor(ccPowDescriptor_t *powDesc); /** * @ingroup dnn * @brief create descriptor of parameters for NonMaxSuppress function * @param [in] point to descriptor of parameters for NonMaxSuppress function * @return ccStatus_t */ ccStatus_t ccCreateNonMaxSuppressionDescriptor(ccNonMaxSuppressionDescriptor_t *nonMaxSuppressionDesc); /** * @ingroup dnn * @brief destroy descriptor of parameters for NonMaxSuppress function * @param [in] point to descriptor of parameters for NonMaxSuppress function * @return ccStatus_t */ ccStatus_t ccDestroyNonMaxSuppressionDescriptor(ccNonMaxSuppressionDescriptor_t *nonMaxSuppressionDesc); ccStatus_t ccTransTensorIncertPads(const ccTensorDescriptor_t xDesc, const void *x, const ccTensorDescriptor_t yDesc, void *y, uint32_t ySizeInBytes, uint32_t boxTypeNum, bool interweave, bool background, uint32_t boxTypeNumMax = 0, bool isScaleVec = false); ccStatus_t ccTransTensorIncertPadsInt32(const ccTensorDescriptor_t xDesc, const void *x, const ccTensorDescriptor_t yDesc, void *y, uint32_t ySizeInBytes, uint32_t boxTypeNum, bool interweave, bool background); ccStatus_t ccTransMskrcnnBbox(const ccTensorDescriptor_t xDesc, const void *x, const ccTensorDescriptor_t yDesc, void *y, uint32_t ySizeInBytes, uint32_t boxTypeNum); ccStatus_t ccSetTensorDescriptorQuantizeParam(ccTensorDescriptor_t tensorDesc, const ccVecQuantizePara_t *vecQuantizePara); ccStatus_t ccGetTensorDescriptorQuantizeParam(const ccTensorDescriptor_t tensorDesc, ccVecQuantizePara_t *vecQuantizePara); /** * @ingroup dnn * @brief init tensor to 4d filter * @param [in|out] filterDesc descriptor of filter * @param [in] format format of filter * @param [in] dataType data type in device * @param [in] k number of output feature maps * @param [in] c number of input feature maps * @param [in] h height of filter * @param [in] w width of filter * @return ccStatus_t */ ccStatus_t ccSetFilter4dDescriptor(ccFilterDescriptor_t filterDesc, ccTensorFormat_t format, ccDataType_t dataType, int32_t k, int32_t c, int32_t h, int32_t w); ccStatus_t ccSetFilter6dDescriptor(ccTensorDescriptor_t filterDesc, ccTensorFormat_t format, ccDataType_t dataType, int32_t c1, int32_t h, int32_t w, int32_t n, int32_t co, int32_t c0); /** * @ingroup dnn * @brief init tensor to Fractal filter * @param [in|out] filterDesc descriptor of filter * @param [in] format format of filter * @param [in] dataType data type in device * @param [in] k number of output feature maps * @param [in] c number of input feature maps * @param [in] h height of filter * @param [in] w width of filter * @return ccStatus_t */ ccStatus_t ccSetFilterFractalDescriptor(ccFilterDescriptor_t filterDesc, ccTensorFormat_t format, ccDataType_t dataType, int32_t k, int32_t c, int32_t h, int32_t w); /** * @ingroup dnn * @brief init tensor to Fractal filter * @param [in|out] filterDesc descriptor of filter * @param [in] format format of filter * @param [in] dataType data type in device * @param [in] k number of output feature maps * @param [in] c number of input feature maps * @param [in] h height of filter * @param [in] w width of filter * @return ccStatus_t */ ccStatus_t ccSetInt8Filter4dDescriptor(ccFilterDescriptor_t filterDesc, ccTensorFormat_t format, ccDataType_t dataType, int32_t k, int32_t c, int32_t h, int32_t w, ccDataType_t outputDataType); /** * @ingroup dnn * @brief read 4d filter * @param [in] filterDesc descriptor of filter * @param [in|out] format point to format of filter * @param [in|out] dataType point to data type in device * @param [in|out] k point to number of output feature maps * @param [in|out] c point to number of input feature maps * @param [in|out] h point to height of filter * @param [in|out] w point to width of filter * @return ccStatus_t */ ccStatus_t ccGetFilterFractalDescriptor(const ccFilterDescriptor_t filterDesc, ccTensorFormat_t *format, ccDataType_t *dataType, int32_t *k, int32_t *c, int32_t *h, int32_t *w); /** * @ingroup dnn * @brief get data size of 4d filter * @param [in] filterDesc descriptor of filter * @param [in|out] size point to data size * @return ccStatus_t */ ccStatus_t ccGetDepthWiseConvFilterSizeInBytes(const ccFilterDescriptor_t filterDesc, int32_t groupNum, uint32_t *size); /** * @ingroup dnn * @brief trans group conv filter to fractal format * @param [in] filterSrcInfo descriptor of input filter * @param [in] filterSrc input data pointer * @param [in] filterDstInfo descriptor of output filter * @param [in|out] filterDst output data pointer * @param [in] group group size * @return ccStatus_t */ ccStatus_t ccTransGroupConvFilter(ccFilterDescriptor_t filterSrcInfo, const void *filterSrc, ccFilterDescriptor_t filterDstInfo, void *filterDst, uint32_t group, uint32_t dstSize); /** * @ingroup dnn * @brief trans conv filter With BoxTypeNuM to fractal format * @param [in] filterSrcInfo descriptor of input filter * @param [in] filterSrc input data pointer * @param [in] filterDstInfo descriptor of output filter * @param [in|out] filterDst output data pointer * @param [in] ySizeInBytes the malloc memory size * @param [in] boxTypeNum the num of boxType * @param [in] inter