更新时间:2021-03-18 GMT+08:00
分享

功能描述

本样例使用TIK方式进行了matmul算子的实现,给定两个输入的张量A和B,进行矩阵相乘,输出结果张量。

说明:此样例未实现算子适配插件,若开发者想在原始框架为caffe的网络模型中使用此自定义算子,需要自定义实现对应的算子适配插件。

分享:

    相关文档

    相关产品

close