Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
5405aa5b
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
5405aa5b
编写于
8月 10, 2020
作者:
Z
zhanyuan
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Remove NNACL's c++ features
上级
33784fef
变更
242
展开全部
隐藏空白更改
内联
并排
Showing
242 changed file
with
683 addition
and
655 deletion
+683
-655
mindspore/lite/CMakeLists.txt
mindspore/lite/CMakeLists.txt
+1
-0
mindspore/lite/src/populate_parameter.cc
mindspore/lite/src/populate_parameter.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/base/fullconnection_base.h
...re/lite/src/runtime/kernel/arm/base/fullconnection_base.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/base/matmul_base.h
mindspore/lite/src/runtime/kernel/arm/base/matmul_base.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/fp32/convolution_1x1.h
mindspore/lite/src/runtime/kernel/arm/fp32/convolution_1x1.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/fp32/matmul.h
mindspore/lite/src/runtime/kernel/arm/fp32/matmul.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/int8/add_int8.cc
mindspore/lite/src/runtime/kernel/arm/int8/add_int8.cc
+1
-0
mindspore/lite/src/runtime/kernel/arm/int8/deconvolution_int8.h
...ore/lite/src/runtime/kernel/arm/int8/deconvolution_int8.h
+2
-1
mindspore/lite/src/runtime/kernel/arm/int8/depth_to_space_int8.h
...re/lite/src/runtime/kernel/arm/int8/depth_to_space_int8.h
+1
-0
mindspore/lite/src/runtime/kernel/arm/int8/fullconnection_int8.cc
...e/lite/src/runtime/kernel/arm/int8/fullconnection_int8.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/int8/hswish_int8.h
mindspore/lite/src/runtime/kernel/arm/int8/hswish_int8.h
+1
-0
mindspore/lite/src/runtime/kernel/arm/int8/matmul_int8.cc
mindspore/lite/src/runtime/kernel/arm/int8/matmul_int8.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/int8/sigmoid_int8.cc
mindspore/lite/src/runtime/kernel/arm/int8/sigmoid_int8.cc
+1
-0
mindspore/lite/src/runtime/kernel/arm/int8/softmax_int8.h
mindspore/lite/src/runtime/kernel/arm/int8/softmax_int8.h
+1
-0
mindspore/lite/src/runtime/kernel/arm/nnacl/activation_grad.h
...spore/lite/src/runtime/kernel/arm/nnacl/activation_grad.h
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/add_int8.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/add_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/add_int8.h
mindspore/lite/src/runtime/kernel/arm/nnacl/add_int8.h
+3
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/arg_min_max.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/arg_min_max.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/arg_min_max.h
mindspore/lite/src/runtime/kernel/arm/nnacl/arg_min_max.h
+1
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/arg_min_max_parameter.h
...lite/src/runtime/kernel/arm/nnacl/arg_min_max_parameter.h
+17
-17
mindspore/lite/src/runtime/kernel/arm/nnacl/arithmetic_common.cc
...re/lite/src/runtime/kernel/arm/nnacl/arithmetic_common.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/arithmetic_common.h
...ore/lite/src/runtime/kernel/arm/nnacl/arithmetic_common.h
+5
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/arithmetic_parameter.h
.../lite/src/runtime/kernel/arm/nnacl/arithmetic_parameter.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/arithmetic_self_parameter.h
.../src/runtime/kernel/arm/nnacl/arithmetic_self_parameter.h
+5
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/batch_to_space.cc
...spore/lite/src/runtime/kernel/arm/nnacl/batch_to_space.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/batch_to_space.h
mindspore/lite/src/runtime/kernel/arm/nnacl/batch_to_space.h
+7
-7
mindspore/lite/src/runtime/kernel/arm/nnacl/common_func.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/common_func.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/common_func.h
mindspore/lite/src/runtime/kernel/arm/nnacl/common_func.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/concat_parameter.h
...pore/lite/src/runtime/kernel/arm/nnacl/concat_parameter.h
+5
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/conv_parameter.h
mindspore/lite/src/runtime/kernel/arm/nnacl/conv_parameter.h
+4
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/crop_parameter.h
mindspore/lite/src/runtime/kernel/arm/nnacl/crop_parameter.h
+5
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/depth_to_space.cc
...spore/lite/src/runtime/kernel/arm/nnacl/depth_to_space.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/depth_to_space.h
mindspore/lite/src/runtime/kernel/arm/nnacl/depth_to_space.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/depth_to_space_parameter.h
...e/src/runtime/kernel/arm/nnacl/depth_to_space_parameter.h
+12
-12
mindspore/lite/src/runtime/kernel/arm/nnacl/errorcode.h
mindspore/lite/src/runtime/kernel/arm/nnacl/errorcode.h
+16
-8
mindspore/lite/src/runtime/kernel/arm/nnacl/flatten.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/flatten.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/flatten.h
mindspore/lite/src/runtime/kernel/arm/nnacl/flatten.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/common_func.h
...pore/lite/src/runtime/kernel/arm/nnacl/fp16/common_func.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_depthwise_fp16.cc
.../src/runtime/kernel/arm/nnacl/fp16/conv_depthwise_fp16.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_depthwise_fp16.h
...e/src/runtime/kernel/arm/nnacl/fp16/conv_depthwise_fp16.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_fp16.cc
...spore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_fp16.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_fp16.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_fp16.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/pack_fp16.cc
...spore/lite/src/runtime/kernel/arm/nnacl/fp16/pack_fp16.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/pack_fp16.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/pack_fp16.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/winograd_transform_fp16.cc
.../runtime/kernel/arm/nnacl/fp16/winograd_transform_fp16.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/winograd_transform_fp16.h
...c/runtime/kernel/arm/nnacl/fp16/winograd_transform_fp16.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/activation.h
...spore/lite/src/runtime/kernel/arm/nnacl/fp32/activation.h
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.cc
...ore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.h
...pore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic.cc
...pore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic.cc
+2
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic.h
...spore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic_self.cc
...lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic_self.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic_self.h
.../lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic_self.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/batchnorm.cc
...spore/lite/src/runtime/kernel/arm/nnacl/fp32/batchnorm.cc
+2
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/batchnorm.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/batchnorm.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/broadcast_to.cc
...re/lite/src/runtime/kernel/arm/nnacl/fp32/broadcast_to.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/broadcast_to.h
...ore/lite/src/runtime/kernel/arm/nnacl/fp32/broadcast_to.h
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/cast.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/cast.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/cast.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/cast.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/common_func.cc
...ore/lite/src/runtime/kernel/arm/nnacl/fp32/common_func.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/common_func.h
...pore/lite/src/runtime/kernel/arm/nnacl/fp32/common_func.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/concat.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/concat.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/concat.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/concat.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv.h
+6
-6
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv_depthwise.cc
.../lite/src/runtime/kernel/arm/nnacl/fp32/conv_depthwise.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv_depthwise.h
...e/lite/src/runtime/kernel/arm/nnacl/fp32/conv_depthwise.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/crop.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/crop.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/crop.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/crop.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/deconv.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/deconv.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/deconv.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/deconv.h
+4
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/elu.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/elu.cc
+3
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/elu.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/elu.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/embedding_lookup.cc
...ite/src/runtime/kernel/arm/nnacl/fp32/embedding_lookup.cc
+3
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/embedding_lookup.h
...lite/src/runtime/kernel/arm/nnacl/fp32/embedding_lookup.h
+9
-9
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/expandDims.cc
...pore/lite/src/runtime/kernel/arm/nnacl/fp32/expandDims.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/expandDims.h
...spore/lite/src/runtime/kernel/arm/nnacl/fp32/expandDims.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/fill.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/fill.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/fill.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/fill.h
+4
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gather.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gather.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gather.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gather.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gatherNd.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gatherNd.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gatherNd.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gatherNd.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/local_response_norm.cc
.../src/runtime/kernel/arm/nnacl/fp32/local_response_norm.cc
+2
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/local_response_norm.h
...e/src/runtime/kernel/arm/nnacl/fp32/local_response_norm.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/lstm.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/lstm.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/lstm.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/lstm.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/matmul.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/matmul.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/matmul.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/matmul.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/one_hot.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/one_hot.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/one_hot.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/one_hot.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pad.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pad.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pad.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pad.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pooling.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pooling.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pooling.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pooling.h
+4
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/range.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/range.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/range.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/range.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/rank.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/rank.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/rank.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/rank.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reduce.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reduce.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reduce.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reduce.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reverse.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reverse.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reverse.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reverse.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/slice.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/slice.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/slice.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/slice.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/softmax.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/softmax.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/softmax.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/softmax.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_batch.cc
.../lite/src/runtime/kernel/arm/nnacl/fp32/space_to_batch.cc
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_batch.h
...e/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_batch.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_depth.cc
.../lite/src/runtime/kernel/arm/nnacl/fp32/space_to_depth.cc
+4
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_depth.h
...e/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_depth.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/stack.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/stack.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/stack.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/stack.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/strassen_matmul.cc
...lite/src/runtime/kernel/arm/nnacl/fp32/strassen_matmul.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/strassen_matmul.h
.../lite/src/runtime/kernel/arm/nnacl/fp32/strassen_matmul.h
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/topk.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/topk.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/topk.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/topk.h
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/unsqueeze.cc
...spore/lite/src/runtime/kernel/arm/nnacl/fp32/unsqueeze.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/unsqueeze.h
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/unsqueeze.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/activation_grad.h
.../src/runtime/kernel/arm/nnacl/fp32_grad/activation_grad.h
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/arithmetic_grad.cc
...src/runtime/kernel/arm/nnacl/fp32_grad/arithmetic_grad.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/batch_norm.cc
...lite/src/runtime/kernel/arm/nnacl/fp32_grad/batch_norm.cc
+11
-11
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/batch_norm.h
.../lite/src/runtime/kernel/arm/nnacl/fp32_grad/batch_norm.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/gemm.cc
...spore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/gemm.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pack_ext.cc
...e/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pack_ext.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pack_ext.h
...re/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pack_ext.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pooling_grad.cc
...te/src/runtime/kernel/arm/nnacl/fp32_grad/pooling_grad.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pooling_grad.h
...ite/src/runtime/kernel/arm/nnacl/fp32_grad/pooling_grad.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/reduce_grad.cc
...ite/src/runtime/kernel/arm/nnacl/fp32_grad/reduce_grad.cc
+2
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/softmax_grad.h
...ite/src/runtime/kernel/arm/nnacl/fp32_grad/softmax_grad.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/fused_batchnorm.cc
...pore/lite/src/runtime/kernel/arm/nnacl/fused_batchnorm.cc
+2
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/fused_batchnorm.h
...spore/lite/src/runtime/kernel/arm/nnacl/fused_batchnorm.h
+3
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arg_min_max_int8.cc
...ite/src/runtime/kernel/arm/nnacl/int8/arg_min_max_int8.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arg_min_max_int8.h
...lite/src/runtime/kernel/arm/nnacl/int8/arg_min_max_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_int8.cc
...lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_int8.h
.../lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_int8.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_self_int8.cc
...src/runtime/kernel/arm/nnacl/int8/arithmetic_self_int8.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_self_int8.h
.../src/runtime/kernel/arm/nnacl/int8/arithmetic_self_int8.h
+3
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/batch_to_space_int8.cc
.../src/runtime/kernel/arm/nnacl/int8/batch_to_space_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/batch_to_space_int8.h
...e/src/runtime/kernel/arm/nnacl/int8/batch_to_space_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/common_func.h
...pore/lite/src/runtime/kernel/arm/nnacl/int8/common_func.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/concat_int8.cc
...ore/lite/src/runtime/kernel/arm/nnacl/int8/concat_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/concat_int8.h
...pore/lite/src/runtime/kernel/arm/nnacl/int8/concat_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_depthwise_int8.cc
.../src/runtime/kernel/arm/nnacl/int8/conv_depthwise_int8.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_depthwise_int8.h
...e/src/runtime/kernel/arm/nnacl/int8/conv_depthwise_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_int8.cc
...spore/lite/src/runtime/kernel/arm/nnacl/int8/conv_int8.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_int8.h
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_int8.h
+6
-6
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/crop_int8.cc
...spore/lite/src/runtime/kernel/arm/nnacl/int8/crop_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/crop_int8.h
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/crop_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/deconv.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/deconv.cc
+2
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/deconv.h
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/deconv.h
+5
-6
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/depth_to_space_int8.cc
.../src/runtime/kernel/arm/nnacl/int8/depth_to_space_int8.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/depth_to_space_int8.h
...e/src/runtime/kernel/arm/nnacl/int8/depth_to_space_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/hswish_int8.cc
...ore/lite/src/runtime/kernel/arm/nnacl/int8/hswish_int8.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/hswish_int8.h
...pore/lite/src/runtime/kernel/arm/nnacl/int8/hswish_int8.h
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/matmul_int8.cc
...ore/lite/src/runtime/kernel/arm/nnacl/int8/matmul_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/matmul_int8.h
...pore/lite/src/runtime/kernel/arm/nnacl/int8/matmul_int8.h
+1
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/mul_int8.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/mul_int8.cc
+4
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/mul_int8.h
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/mul_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pad.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pad.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pad.h
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pad.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pooling_int8.cc
...re/lite/src/runtime/kernel/arm/nnacl/int8/pooling_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pooling_int8.h
...ore/lite/src/runtime/kernel/arm/nnacl/int8/pooling_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/prelu_int8.cc
...pore/lite/src/runtime/kernel/arm/nnacl/int8/prelu_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/prelu_int8.h
...spore/lite/src/runtime/kernel/arm/nnacl/int8/prelu_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/quant_dtype_cast.cc
...ite/src/runtime/kernel/arm/nnacl/int8/quant_dtype_cast.cc
+3
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/quant_dtype_cast.h
...lite/src/runtime/kernel/arm/nnacl/int8/quant_dtype_cast.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/relux_int8.h
...spore/lite/src/runtime/kernel/arm/nnacl/int8/relux_int8.h
+6
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/reshape_int8.cc
...re/lite/src/runtime/kernel/arm/nnacl/int8/reshape_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/reshape_int8.h
...ore/lite/src/runtime/kernel/arm/nnacl/int8/reshape_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/sigmoid_int8.cc
...re/lite/src/runtime/kernel/arm/nnacl/int8/sigmoid_int8.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/sigmoid_int8.h
...ore/lite/src/runtime/kernel/arm/nnacl/int8/sigmoid_int8.h
+5
-5
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/softmax_int8.cc
...re/lite/src/runtime/kernel/arm/nnacl/int8/softmax_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/softmax_int8.h
...ore/lite/src/runtime/kernel/arm/nnacl/int8/softmax_int8.h
+3
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/split_int8.cc
...pore/lite/src/runtime/kernel/arm/nnacl/int8/split_int8.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/split_int8.h
...spore/lite/src/runtime/kernel/arm/nnacl/int8/split_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/squeeze_int8.cc
...re/lite/src/runtime/kernel/arm/nnacl/int8/squeeze_int8.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/squeeze_int8.h
...ore/lite/src/runtime/kernel/arm/nnacl/int8/squeeze_int8.h
+2
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/topk_int8.cc
...spore/lite/src/runtime/kernel/arm/nnacl/int8/topk_int8.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/topk_int8.h
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/topk_int8.h
+4
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/unsqueeze_int8.cc
.../lite/src/runtime/kernel/arm/nnacl/int8/unsqueeze_int8.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/unsqueeze_int8.h
...e/lite/src/runtime/kernel/arm/nnacl/int8/unsqueeze_int8.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/matmul_parameter.h
...pore/lite/src/runtime/kernel/arm/nnacl/matmul_parameter.h
+4
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/mul_parameter.h
mindspore/lite/src/runtime/kernel/arm/nnacl/mul_parameter.h
+4
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/nnacl_utils.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/nnacl_utils.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/op_base.h
mindspore/lite/src/runtime/kernel/arm/nnacl/op_base.h
+6
-6
mindspore/lite/src/runtime/kernel/arm/nnacl/optimized_kernel.h
...pore/lite/src/runtime/kernel/arm/nnacl/optimized_kernel.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/pack.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/pack.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/pack.h
mindspore/lite/src/runtime/kernel/arm/nnacl/pack.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/pad_parameter.h
mindspore/lite/src/runtime/kernel/arm/nnacl/pad_parameter.h
+4
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/power.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/power.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/power.h
mindspore/lite/src/runtime/kernel/arm/nnacl/power.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/prelu.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/prelu.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/prelu.h
mindspore/lite/src/runtime/kernel/arm/nnacl/prelu.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/prelu_parameter.h
...spore/lite/src/runtime/kernel/arm/nnacl/prelu_parameter.h
+5
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/prior_box.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/prior_box.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/prior_box.h
mindspore/lite/src/runtime/kernel/arm/nnacl/prior_box.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/quantization/quantize.cc
...ite/src/runtime/kernel/arm/nnacl/quantization/quantize.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/quantization/quantize.h
...lite/src/runtime/kernel/arm/nnacl/quantization/quantize.h
+53
-54
mindspore/lite/src/runtime/kernel/arm/nnacl/reshape.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/reshape.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/reshape.h
mindspore/lite/src/runtime/kernel/arm/nnacl/reshape.h
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/reshape_parameter.h
...ore/lite/src/runtime/kernel/arm/nnacl/reshape_parameter.h
+4
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/resize.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/resize.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/resize.h
mindspore/lite/src/runtime/kernel/arm/nnacl/resize.h
+4
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/reverse_sequence.cc
...ore/lite/src/runtime/kernel/arm/nnacl/reverse_sequence.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/reverse_sequence.h
...pore/lite/src/runtime/kernel/arm/nnacl/reverse_sequence.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/scale.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/scale.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/scale.h
mindspore/lite/src/runtime/kernel/arm/nnacl/scale.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/scatter_nd.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/scatter_nd.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/scatter_nd.h
mindspore/lite/src/runtime/kernel/arm/nnacl/scatter_nd.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/shape.h
mindspore/lite/src/runtime/kernel/arm/nnacl/shape.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/sigmoid_parameter.h
...ore/lite/src/runtime/kernel/arm/nnacl/sigmoid_parameter.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/softmax_parameter.h
...ore/lite/src/runtime/kernel/arm/nnacl/softmax_parameter.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/sparse_to_dense.cc
...pore/lite/src/runtime/kernel/arm/nnacl/sparse_to_dense.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/sparse_to_dense.h
...spore/lite/src/runtime/kernel/arm/nnacl/sparse_to_dense.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/split.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/split.cc
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/split.h
mindspore/lite/src/runtime/kernel/arm/nnacl/split.h
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/split_parameter.h
...spore/lite/src/runtime/kernel/arm/nnacl/split_parameter.h
+4
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/squeeze.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/squeeze.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/squeeze.h
mindspore/lite/src/runtime/kernel/arm/nnacl/squeeze.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/squeeze_parameter.h
...ore/lite/src/runtime/kernel/arm/nnacl/squeeze_parameter.h
+4
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/strassen_matmul.h
...spore/lite/src/runtime/kernel/arm/nnacl/strassen_matmul.h
+9
-10
mindspore/lite/src/runtime/kernel/arm/nnacl/strided_slice.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/strided_slice.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/strided_slice.h
mindspore/lite/src/runtime/kernel/arm/nnacl/strided_slice.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/tile.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/tile.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/tile.h
mindspore/lite/src/runtime/kernel/arm/nnacl/tile.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/transpose.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/transpose.cc
+2
-2
mindspore/lite/src/runtime/kernel/arm/nnacl/transpose.h
mindspore/lite/src/runtime/kernel/arm/nnacl/transpose.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/unique.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/unique.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/unique.h
mindspore/lite/src/runtime/kernel/arm/nnacl/unique.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/unsqueeze_parameter.h
...e/lite/src/runtime/kernel/arm/nnacl/unsqueeze_parameter.h
+4
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/unstack.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/unstack.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/unstack.h
mindspore/lite/src/runtime/kernel/arm/nnacl/unstack.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/where.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/where.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/where.h
mindspore/lite/src/runtime/kernel/arm/nnacl/where.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_transform.cc
...e/lite/src/runtime/kernel/arm/nnacl/winograd_transform.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_transform.h
...re/lite/src/runtime/kernel/arm/nnacl/winograd_transform.h
+4
-4
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_utils.cc
...spore/lite/src/runtime/kernel/arm/nnacl/winograd_utils.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_utils.h
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_utils.h
+3
-3
mindspore/lite/src/runtime/kernel/arm/nnacl/zeroslike.cc
mindspore/lite/src/runtime/kernel/arm/nnacl/zeroslike.cc
+1
-1
mindspore/lite/src/runtime/kernel/arm/nnacl/zeroslike.h
mindspore/lite/src/runtime/kernel/arm/nnacl/zeroslike.h
+1
-1
mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/conv1x1_fp32_tests.cc
...test/ut/src/runtime/kernel/arm/fp32/conv1x1_fp32_tests.cc
+1
-1
未找到文件。
mindspore/lite/CMakeLists.txt
浏览文件 @
5405aa5b
...
@@ -13,6 +13,7 @@ include_directories(${TOP_DIR})
...
@@ -13,6 +13,7 @@ include_directories(${TOP_DIR})
include_directories
(
${
CORE_DIR
}
)
include_directories
(
${
CORE_DIR
}
)
include_directories
(
${
CCSRC_DIR
}
)
include_directories
(
${
CCSRC_DIR
}
)
include_directories
(
${
CMAKE_CURRENT_SOURCE_DIR
}
)
include_directories
(
${
CMAKE_CURRENT_SOURCE_DIR
}
)
include_directories
(
${
CMAKE_CURRENT_SOURCE_DIR
}
/src/runtime/kernel/arm
)
include_directories
(
${
TOP_DIR
}
/third_party
)
include_directories
(
${
TOP_DIR
}
/third_party
)
include_directories
(
${
TOP_DIR
}
/third_party/flatbuffers/include
)
include_directories
(
${
TOP_DIR
}
/third_party/flatbuffers/include
)
...
...
mindspore/lite/src/populate_parameter.cc
浏览文件 @
5405aa5b
...
@@ -31,7 +31,7 @@
...
@@ -31,7 +31,7 @@
#include "src/runtime/kernel/arm/nnacl/depth_to_space.h"
#include "src/runtime/kernel/arm/nnacl/depth_to_space.h"
#include "src/runtime/kernel/arm/nnacl/conv_parameter.h"
#include "src/runtime/kernel/arm/nnacl/conv_parameter.h"
#include "src/runtime/kernel/arm/nnacl/fp32/pooling.h"
#include "src/runtime/kernel/arm/nnacl/fp32/pooling.h"
#include "src/runtime/kernel/arm/nnacl/matmul.h"
#include "src/runtime/kernel/arm/nnacl/matmul
_parameter
.h"
#include "src/runtime/kernel/arm/nnacl/softmax_parameter.h"
#include "src/runtime/kernel/arm/nnacl/softmax_parameter.h"
#include "src/runtime/kernel/arm/nnacl/tile.h"
#include "src/runtime/kernel/arm/nnacl/tile.h"
#include "src/runtime/kernel/arm/nnacl/fp32/topk.h"
#include "src/runtime/kernel/arm/nnacl/fp32/topk.h"
...
...
mindspore/lite/src/runtime/kernel/arm/base/fullconnection_base.h
浏览文件 @
5405aa5b
...
@@ -20,7 +20,7 @@
...
@@ -20,7 +20,7 @@
#include <vector>
#include <vector>
#include "src/lite_kernel.h"
#include "src/lite_kernel.h"
#include "include/context.h"
#include "include/context.h"
#include "src/runtime/kernel/arm/nnacl/matmul.h"
#include "src/runtime/kernel/arm/nnacl/matmul
_parameter
.h"
using
mindspore
::
lite
::
Context
;
using
mindspore
::
lite
::
Context
;
...
...
mindspore/lite/src/runtime/kernel/arm/base/matmul_base.h
浏览文件 @
5405aa5b
...
@@ -20,7 +20,7 @@
...
@@ -20,7 +20,7 @@
#include <vector>
#include <vector>
#include "src/lite_kernel.h"
#include "src/lite_kernel.h"
#include "include/context.h"
#include "include/context.h"
#include "src/runtime/kernel/arm/nnacl/matmul.h"
#include "src/runtime/kernel/arm/nnacl/matmul
_parameter
.h"
using
mindspore
::
lite
::
Context
;
using
mindspore
::
lite
::
Context
;
...
...
mindspore/lite/src/runtime/kernel/arm/fp32/convolution_1x1.h
浏览文件 @
5405aa5b
...
@@ -27,7 +27,7 @@
...
@@ -27,7 +27,7 @@
#include "src/runtime/kernel/arm/base/layout_transform.h"
#include "src/runtime/kernel/arm/base/layout_transform.h"
#include "src/runtime/kernel/arm/nnacl/fp32/conv.h"
#include "src/runtime/kernel/arm/nnacl/fp32/conv.h"
#include "src/runtime/kernel/arm/nnacl/fp32/common_func.h"
#include "src/runtime/kernel/arm/nnacl/fp32/common_func.h"
#include "src/runtime/kernel/arm/nnacl/matmul.h"
#include "src/runtime/kernel/arm/nnacl/matmul
_parameter
.h"
#include "src/runtime/kernel/arm/nnacl/fp32/matmul.h"
#include "src/runtime/kernel/arm/nnacl/fp32/matmul.h"
namespace
mindspore
::
kernel
{
namespace
mindspore
::
kernel
{
...
...
mindspore/lite/src/runtime/kernel/arm/fp32/matmul.h
浏览文件 @
5405aa5b
...
@@ -19,7 +19,7 @@
...
@@ -19,7 +19,7 @@
#include <vector>
#include <vector>
#include "src/lite_kernel.h"
#include "src/lite_kernel.h"
#include "src/runtime/kernel/arm/nnacl/matmul.h"
#include "src/runtime/kernel/arm/nnacl/matmul
_parameter
.h"
#include "src/runtime/kernel/arm/base/matmul_base.h"
#include "src/runtime/kernel/arm/base/matmul_base.h"
namespace
mindspore
::
kernel
{
namespace
mindspore
::
kernel
{
...
...
mindspore/lite/src/runtime/kernel/arm/int8/add_int8.cc
浏览文件 @
5405aa5b
...
@@ -18,6 +18,7 @@
...
@@ -18,6 +18,7 @@
#include <limits>
#include <limits>
#include <algorithm>
#include <algorithm>
#include "src/runtime/kernel/arm/nnacl/arithmetic_common.h"
#include "src/runtime/kernel/arm/nnacl/arithmetic_common.h"
#include "src/runtime/kernel/arm/nnacl/quantization/quantize.h"
#include "src/runtime/runtime_api.h"
#include "src/runtime/runtime_api.h"
#include "src/kernel_registry.h"
#include "src/kernel_registry.h"
#include "include/errorcode.h"
#include "include/errorcode.h"
...
...
mindspore/lite/src/runtime/kernel/arm/int8/deconvolution_int8.h
浏览文件 @
5405aa5b
...
@@ -21,8 +21,9 @@
...
@@ -21,8 +21,9 @@
#include "schema/model_generated.h"
#include "schema/model_generated.h"
#include "src/kernel_registry.h"
#include "src/kernel_registry.h"
#include "include/errorcode.h"
#include "include/errorcode.h"
#include "src/runtime/kernel/arm/nnacl/matmul_parameter.h"
#include "src/runtime/kernel/arm/nnacl/int8/deconv.h"
#include "src/runtime/kernel/arm/nnacl/int8/deconv.h"
#include "src/runtime/kernel/arm/nnacl/int8/matmul.h"
#include "src/runtime/kernel/arm/nnacl/int8/matmul
_int8
.h"
#include "src/runtime/kernel/arm/base/layout_transform.h"
#include "src/runtime/kernel/arm/base/layout_transform.h"
#include "src/runtime/kernel/arm/base/convolution_base.h"
#include "src/runtime/kernel/arm/base/convolution_base.h"
#include "src/runtime/kernel/arm/nnacl/arithmetic_common.h"
#include "src/runtime/kernel/arm/nnacl/arithmetic_common.h"
...
...
mindspore/lite/src/runtime/kernel/arm/int8/depth_to_space_int8.h
浏览文件 @
5405aa5b
...
@@ -18,6 +18,7 @@
...
@@ -18,6 +18,7 @@
#include <vector>
#include <vector>
#include "src/runtime/kernel/arm/base/depth_to_space_base.h"
#include "src/runtime/kernel/arm/base/depth_to_space_base.h"
#include "src/runtime/kernel/arm/nnacl/quantization/quantize.h"
namespace
mindspore
::
kernel
{
namespace
mindspore
::
kernel
{
class
DepthToSpaceInt8CPUKernel
:
public
DepthToSpaceBaseCPUKernel
{
class
DepthToSpaceInt8CPUKernel
:
public
DepthToSpaceBaseCPUKernel
{
...
...
mindspore/lite/src/runtime/kernel/arm/int8/fullconnection_int8.cc
浏览文件 @
5405aa5b
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
*/
*/
#include "src/runtime/kernel/arm/int8/fullconnection_int8.h"
#include "src/runtime/kernel/arm/int8/fullconnection_int8.h"
#include "src/runtime/kernel/arm/nnacl/int8/matmul.h"
#include "src/runtime/kernel/arm/nnacl/int8/matmul
_int8
.h"
#include "src/runtime/kernel/arm/nnacl/common_func.h"
#include "src/runtime/kernel/arm/nnacl/common_func.h"
#include "src/runtime/runtime_api.h"
#include "src/runtime/runtime_api.h"
#include "include/errorcode.h"
#include "include/errorcode.h"
...
...
mindspore/lite/src/runtime/kernel/arm/int8/hswish_int8.h
浏览文件 @
5405aa5b
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include "src/lite_kernel.h"
#include "src/lite_kernel.h"
#include "src/runtime/kernel/arm/nnacl/int8/hswish_int8.h"
#include "src/runtime/kernel/arm/nnacl/int8/hswish_int8.h"
#include "src/runtime/kernel/arm/nnacl/quantization/quantize.h"
namespace
mindspore
::
kernel
{
namespace
mindspore
::
kernel
{
class
HswishInt8CPUKernel
:
public
LiteKernel
{
class
HswishInt8CPUKernel
:
public
LiteKernel
{
...
...
mindspore/lite/src/runtime/kernel/arm/int8/matmul_int8.cc
浏览文件 @
5405aa5b
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
*/
*/
#include "src/runtime/kernel/arm/int8/matmul_int8.h"
#include "src/runtime/kernel/arm/int8/matmul_int8.h"
#include "src/runtime/kernel/arm/nnacl/int8/matmul.h"
#include "src/runtime/kernel/arm/nnacl/int8/matmul
_int8
.h"
#include "src/runtime/kernel/arm/nnacl/common_func.h"
#include "src/runtime/kernel/arm/nnacl/common_func.h"
#include "src/runtime/runtime_api.h"
#include "src/runtime/runtime_api.h"
#include "include/errorcode.h"
#include "include/errorcode.h"
...
...
mindspore/lite/src/runtime/kernel/arm/int8/sigmoid_int8.cc
浏览文件 @
5405aa5b
...
@@ -17,6 +17,7 @@
...
@@ -17,6 +17,7 @@
#include "src/runtime/kernel/arm/int8/sigmoid_int8.h"
#include "src/runtime/kernel/arm/int8/sigmoid_int8.h"
#include <limits>
#include <limits>
#include "src/runtime/kernel/arm/nnacl/int8/sigmoid_int8.h"
#include "src/runtime/kernel/arm/nnacl/int8/sigmoid_int8.h"
#include "src/runtime/kernel/arm/nnacl/quantization/quantize.h"
#include "schema/model_generated.h"
#include "schema/model_generated.h"
#include "src/kernel_registry.h"
#include "src/kernel_registry.h"
#include "src/runtime/runtime_api.h"
#include "src/runtime/runtime_api.h"
...
...
mindspore/lite/src/runtime/kernel/arm/int8/softmax_int8.h
浏览文件 @
5405aa5b
...
@@ -19,6 +19,7 @@
...
@@ -19,6 +19,7 @@
#include <vector>
#include <vector>
#include "src/runtime/kernel/arm/base/softmax_base.h"
#include "src/runtime/kernel/arm/base/softmax_base.h"
#include "src/runtime/kernel/arm/nnacl/quantization/quantize.h"
namespace
mindspore
::
kernel
{
namespace
mindspore
::
kernel
{
class
SoftmaxInt8CPUKernel
:
public
SoftmaxBaseCPUKernel
{
class
SoftmaxInt8CPUKernel
:
public
SoftmaxBaseCPUKernel
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/activation_grad.h
浏览文件 @
5405aa5b
...
@@ -17,15 +17,15 @@
...
@@ -17,15 +17,15 @@
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ACTIVATION_GRAD_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ACTIVATION_GRAD_H_
#include <math.h>
#include <math.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/fp32/arithmetic.h"
#include "nnacl/fp32/arithmetic.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
struct
ActivationGradParameter
{
typedef
struct
ActivationGradParameter
{
OpParameter
op_parameter
{};
OpParameter
op_parameter
{};
int
type_
;
int
type_
;
float
alpha_
{
0
.
01
};
float
alpha_
{
0
.
01
};
};
}
ActivationGradParameter
;
inline
int
ReluGrad
(
float
*
src0
,
float
*
src1
,
int
length
,
float
*
dst
)
{
inline
int
ReluGrad
(
float
*
src0
,
float
*
src1
,
int
length
,
float
*
dst
)
{
for
(
int
i
=
0
;
i
<
length
;
++
i
)
{
for
(
int
i
=
0
;
i
<
length
;
++
i
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/add_int8.cc
浏览文件 @
5405aa5b
...
@@ -14,11 +14,11 @@
...
@@ -14,11 +14,11 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/add_int8.h"
#include "nnacl/add_int8.h"
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/quantization/fixed_point.h"
#include "nnacl/quantization/fixed_point.h"
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
int16x8_t
LoadAndAddOffset
(
int8_t
*
data
,
int
index
,
int
offset
)
{
int16x8_t
LoadAndAddOffset
(
int8_t
*
data
,
int
index
,
int
offset
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/add_int8.h
浏览文件 @
5405aa5b
...
@@ -17,9 +17,9 @@
...
@@ -17,9 +17,9 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ADD_INT8_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ADD_INT8_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ADD_INT8_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ADD_INT8_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
AddQuantParameter
{
typedef
struct
AddQuantParameter
{
int
input0_offset_
;
int
input0_offset_
;
int
input1_offset_
;
int
input1_offset_
;
int
output_offset_
;
int
output_offset_
;
...
@@ -40,7 +40,7 @@ struct AddQuantParameter {
...
@@ -40,7 +40,7 @@ struct AddQuantParameter {
int
right_shift1_
;
int
right_shift1_
;
int
left_shift_out_
;
int
left_shift_out_
;
int
right_shift_out_
;
int
right_shift_out_
;
};
}
AddQuantParameter
;
void
AddInt8
(
int8_t
*
input0_data
,
int8_t
*
input1_data
,
int8_t
*
output_data
,
int64_t
real_dst_count
,
void
AddInt8
(
int8_t
*
input0_data
,
int8_t
*
input1_data
,
int8_t
*
output_data
,
int64_t
real_dst_count
,
AddQuantParameter
*
para
);
AddQuantParameter
*
para
);
...
@@ -51,4 +51,3 @@ int16x8_t LoadAndAddOffset(int8_t *data, int index, int offset);
...
@@ -51,4 +51,3 @@ int16x8_t LoadAndAddOffset(int8_t *data, int index, int offset);
#endif
#endif
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ADD_INT8_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ADD_INT8_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/arg_min_max.cc
浏览文件 @
5405aa5b
...
@@ -13,8 +13,8 @@
...
@@ -13,8 +13,8 @@
* See the License for the specific language governing permissions and
* See the License for the specific language governing permissions and
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/arg_min_max.h"
#include "nnacl/arg_min_max.h"
#include "
src/runtime/kernel/arm/
nnacl/fp32/arg_min_max.h"
#include "nnacl/fp32/arg_min_max.h"
#define FLOAT_DATA_TYPE 43
#define FLOAT_DATA_TYPE 43
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/arg_min_max.h
浏览文件 @
5405aa5b
...
@@ -16,9 +16,7 @@
...
@@ -16,9 +16,7 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_H_
#include "src/runtime/kernel/arm/nnacl/op_base.h"
#include "nnacl/arg_min_max_parameter.h"
#include "src/runtime/kernel/arm/nnacl/arg_min_max_parameter.h"
#include "src/runtime/kernel/arm/nnacl/quantization/quantize.h"
void
ArgMinMax
(
const
void
*
input
,
void
*
output
,
const
int
*
in_shape
,
ArgMinMaxParameter
*
param
);
void
ArgMinMax
(
const
void
*
input
,
void
*
output
,
const
int
*
in_shape
,
ArgMinMaxParameter
*
param
);
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/arg_min_max_parameter.h
浏览文件 @
5405aa5b
...
@@ -17,30 +17,30 @@
...
@@ -17,30 +17,30 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_PARAMETER_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_PARAMETER_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
ArgElement
{
typedef
struct
ArgElement
{
uint32_t
index_
;
uint32_t
index_
;
union
ArgData
{
union
ArgData
{
int8_t
i8_data_
;
int8_t
i8_data_
;
int32_t
i_data_
;
int32_t
i_data_
;
float
f_data_
;
float
f_data_
;
}
data_
;
}
data_
;
};
}
ArgElement
;
struct
ArgMinMaxParameter
{
typedef
struct
ArgMinMaxParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
bool
out_value_
;
bool
out_value_
;
bool
keep_dims_
;
bool
keep_dims_
;
bool
get_max_
;
bool
get_max_
;
int32_t
axis_
;
int32_t
axis_
;
int32_t
topk_
;
int32_t
topk_
;
int32_t
axis_type_
;
int32_t
axis_type_
;
int32_t
dims_size_
;
int32_t
dims_size_
;
int32_t
data_type_
;
// equals to type_id
int32_t
data_type_
;
// equals to type_id
int32_t
in_strides_
[
DIMENSION_4D
];
int32_t
in_strides_
[
DIMENSION_4D
];
int32_t
out_strides_
[
DIMENSION_4D
];
int32_t
out_strides_
[
DIMENSION_4D
];
ArgElement
*
arg_elements_
;
ArgElement
*
arg_elements_
;
};
}
ArgMinMaxParameter
;
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_PARAMETER_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARG_MIN_MAX_PARAMETER_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/arithmetic_common.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/arithmetic_common.h"
#include "nnacl/arithmetic_common.h"
void
TileOneDimension
(
float
*
inData
,
float
*
outData
,
int
dim
,
size_t
ndim
,
int
*
inShape
,
int
*
inStrides
,
void
TileOneDimension
(
float
*
inData
,
float
*
outData
,
int
dim
,
size_t
ndim
,
int
*
inShape
,
int
*
inStrides
,
int
*
outStrides
,
int
*
multiple
)
{
int
*
outStrides
,
int
*
multiple
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/arithmetic_common.h
浏览文件 @
5405aa5b
...
@@ -20,10 +20,10 @@
...
@@ -20,10 +20,10 @@
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/arithmetic_common.h"
#include "nnacl/arithmetic_common.h"
struct
ArithmeticParameter
{
typedef
struct
ArithmeticParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
bool
broadcasting_
;
bool
broadcasting_
;
size_t
ndim_
;
size_t
ndim_
;
...
@@ -38,7 +38,8 @@ struct ArithmeticParameter {
...
@@ -38,7 +38,8 @@ struct ArithmeticParameter {
int
multiples0_
[
5
];
int
multiples0_
[
5
];
int
multiples1_
[
5
];
int
multiples1_
[
5
];
};
}
ArithmeticParameter
;
void
TileOneDimension
(
float
*
inData
,
float
*
outData
,
int
dim
,
size_t
ndim
,
int
*
inShape
,
int
*
inStrides
,
void
TileOneDimension
(
float
*
inData
,
float
*
outData
,
int
dim
,
size_t
ndim
,
int
*
inShape
,
int
*
inStrides
,
int
*
outStrides
,
int
*
multiple
);
int
*
outStrides
,
int
*
multiple
);
void
ComputeStrides
(
int
*
shape
,
int
*
strides
,
int
ndim
);
void
ComputeStrides
(
int
*
shape
,
int
*
strides
,
int
ndim
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/arithmetic_parameter.h
浏览文件 @
5405aa5b
...
@@ -17,7 +17,7 @@
...
@@ -17,7 +17,7 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARTITHMETIC_PARAMETER_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARTITHMETIC_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARTITHMETIC_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARTITHMETIC_PARAMETER_H_
#include "
src/runtime/kernel/arm/
nnacl/op_attribute.h"
#include "nnacl/op_attribute.h"
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/arithmetic_self_parameter.h
浏览文件 @
5405aa5b
...
@@ -17,13 +17,14 @@
...
@@ -17,13 +17,14 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARITHMETIC_SELF_PARAMETER_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARITHMETIC_SELF_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARITHMETIC_SELF_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARITHMETIC_SELF_PARAMETER_H_
#include "src/runtime/kernel/arm/nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "src/runtime/kernel/arm/nnacl/errorcode.h"
#include "nnacl/errorcode.h"
#include "nnacl/quantization/quantize.h"
// For Abs, Cos, Exp, Log, Square, Sqrt, Rsqrt ops.
// For Abs, Cos, Exp, Log, Square, Sqrt, Rsqrt ops.
struct
ArithmeticSelfParameter
{
typedef
struct
ArithmeticSelfParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
ArithSelfQuantArg
quant_arg_
;
ArithSelfQuantArg
quant_arg_
;
};
}
ArithmeticSelfParameter
;
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARITHMETIC_SELF_PARAMETER_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ARITHMETIC_SELF_PARAMETER_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/batch_to_space.cc
浏览文件 @
5405aa5b
...
@@ -14,8 +14,8 @@
...
@@ -14,8 +14,8 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/batch_to_space.h"
#include "nnacl/batch_to_space.h"
#include "
src/runtime/kernel/arm/
nnacl/arithmetic_common.h"
#include "nnacl/arithmetic_common.h"
void
BatchToSpaceNoCropForNHWC
(
const
void
*
input
,
void
*
output
,
const
int
*
in_shape
,
int
out_n
,
const
int
*
block
,
void
BatchToSpaceNoCropForNHWC
(
const
void
*
input
,
void
*
output
,
const
int
*
in_shape
,
int
out_n
,
const
int
*
block
,
int
data_size
)
{
int
data_size
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/batch_to_space.h
浏览文件 @
5405aa5b
...
@@ -15,16 +15,16 @@
...
@@ -15,16 +15,16 @@
*/
*/
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_BATCH_TO_SPACE_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_BATCH_TO_SPACE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_BATCH_TO_SPACE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_BATCH_TO_SPACE_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#define BATCH_TO_SPACE_BLOCK_SHAPE_SIZE 2
#define BATCH_TO_SPACE_BLOCK_SHAPE_SIZE 2
#define BATCH_TO_SPACE_CROPS_SIZE
4
#define BATCH_TO_SPACE_CROPS_SIZE 4
struct
BatchToSpaceParameter
{
typedef
struct
BatchToSpaceParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int32_t
block_shape_
[
BATCH_TO_SPACE_BLOCK_SHAPE_SIZE
];
int32_t
block_shape_
[
BATCH_TO_SPACE_BLOCK_SHAPE_SIZE
];
int32_t
crops_
[
BATCH_TO_SPACE_CROPS_SIZE
];
int32_t
crops_
[
BATCH_TO_SPACE_CROPS_SIZE
];
};
}
BatchToSpaceParameter
;
void
BatchToSpaceNoCropForNHWC
(
const
void
*
input
,
void
*
output
,
const
int
*
in_shape
,
int
out_n
,
const
int
*
block
,
void
BatchToSpaceNoCropForNHWC
(
const
void
*
input
,
void
*
output
,
const
int
*
in_shape
,
int
out_n
,
const
int
*
block
,
int
data_size
);
int
data_size
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/common_func.cc
浏览文件 @
5405aa5b
...
@@ -14,8 +14,8 @@
...
@@ -14,8 +14,8 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/common_func.h"
#include "nnacl/common_func.h"
#include "
src/runtime/kernel/arm/
nnacl/quantization/fixed_point.h"
#include "nnacl/quantization/fixed_point.h"
#ifndef ENABLE_ARM64
#ifndef ENABLE_ARM64
void
IndirectGemmFp32
(
float
*
output
,
const
float
*
input
,
const
float
*
weight
,
const
float
*
bias
,
size_t
step
,
int
ic4
,
void
IndirectGemmFp32
(
float
*
output
,
const
float
*
input
,
const
float
*
weight
,
const
float
*
bias
,
size_t
step
,
int
ic4
,
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/common_func.h
浏览文件 @
5405aa5b
...
@@ -20,8 +20,8 @@
...
@@ -20,8 +20,8 @@
#include <stdint.h>
#include <stdint.h>
#include <stdio.h>
#include <stdio.h>
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
#ifdef __cplusplus
#ifdef __cplusplus
extern
"C"
{
extern
"C"
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/concat_parameter.h
浏览文件 @
5405aa5b
...
@@ -17,8 +17,10 @@
...
@@ -17,8 +17,10 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CONCAT_PARAMETER_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CONCAT_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CONCAT_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CONCAT_PARAMETER_H_
#include "src/runtime/kernel/arm/nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
ConcatParameter
{
#include "nnacl/quantization/quantize.h"
typedef
struct
ConcatParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
ConcatQuantArg
quant_arg_
;
ConcatQuantArg
quant_arg_
;
int
axis_
;
int
axis_
;
...
@@ -28,7 +30,6 @@ struct ConcatParameter {
...
@@ -28,7 +30,6 @@ struct ConcatParameter {
const
int
*
output_shapes_
;
const
int
*
output_shapes_
;
int64_t
after_axis_size
;
int64_t
after_axis_size
;
int64_t
count_unit_
;
int64_t
count_unit_
;
};
}
ConcatParameter
;
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CONCAT_PARAMETER_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CONCAT_PARAMETER_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/conv_parameter.h
浏览文件 @
5405aa5b
...
@@ -20,10 +20,10 @@
...
@@ -20,10 +20,10 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/quantization/quantize.h"
#include "nnacl/quantization/quantize.h"
struct
ConvParameter
{
typedef
struct
ConvParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
ConvQuantArg
conv_quant_arg_
;
ConvQuantArg
conv_quant_arg_
;
int
kernel_h_
;
int
kernel_h_
;
...
@@ -53,7 +53,6 @@ struct ConvParameter {
...
@@ -53,7 +53,6 @@ struct ConvParameter {
int
output_unit_
;
int
output_unit_
;
bool
is_relu_
;
bool
is_relu_
;
bool
is_relu6_
;
bool
is_relu6_
;
};
}
ConvParameter
;
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CONV_PARAMETER_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CONV_PARAMETER_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/crop_parameter.h
浏览文件 @
5405aa5b
...
@@ -16,11 +16,13 @@
...
@@ -16,11 +16,13 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CROP_PARAMETER_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CROP_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CROP_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CROP_PARAMETER_H_
#include "src/runtime/kernel/arm/nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "nnacl/quantization/quantize.h"
#define CROP_OFFSET_MAX_SIZE 4
#define CROP_OFFSET_MAX_SIZE 4
struct
CropParameter
{
typedef
struct
CropParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
CropQuantArg
quant_arg
;
CropQuantArg
quant_arg
;
int
thread_count_
;
int
thread_count_
;
...
@@ -32,6 +34,6 @@ struct CropParameter {
...
@@ -32,6 +34,6 @@ struct CropParameter {
const
int
*
in_shape_
;
const
int
*
in_shape_
;
const
int
*
out_shape_
;
const
int
*
out_shape_
;
int
input_dim_
;
int
input_dim_
;
};
}
CropParameter
;
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CROP_PARAMETER_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_CROP_PARAMETER_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/depth_to_space.cc
浏览文件 @
5405aa5b
...
@@ -13,7 +13,7 @@
...
@@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* See the License for the specific language governing permissions and
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/depth_to_space.h"
#include "nnacl/depth_to_space.h"
#include <string.h>
#include <string.h>
void
DepthToSpaceForNHWC
(
const
void
*
input
,
void
*
output
,
int
*
in_shape
,
DepthToSpaceParameter
*
param
)
{
void
DepthToSpaceForNHWC
(
const
void
*
input
,
void
*
output
,
int
*
in_shape
,
DepthToSpaceParameter
*
param
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/depth_to_space.h
浏览文件 @
5405aa5b
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
*/
*/
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_H_
#include "
src/runtime/kernel/arm/
nnacl/depth_to_space_parameter.h"
#include "nnacl/depth_to_space_parameter.h"
void
DepthToSpaceForNHWC
(
const
void
*
input
,
void
*
output
,
int
*
in_shape
,
DepthToSpaceParameter
*
param
);
void
DepthToSpaceForNHWC
(
const
void
*
input
,
void
*
output
,
int
*
in_shape
,
DepthToSpaceParameter
*
param
);
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/depth_to_space_parameter.h
浏览文件 @
5405aa5b
...
@@ -15,18 +15,18 @@
...
@@ -15,18 +15,18 @@
*/
*/
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_PARAMETER_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_PARAMETER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_PARAMETER_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
DepthToSpaceParameter
{
typedef
struct
DepthToSpaceParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int32_t
block_size_
;
int32_t
block_size_
;
int32_t
in_stride_dim0_
;
int32_t
in_stride_dim0_
;
int32_t
in_stride_dim1_
;
int32_t
in_stride_dim1_
;
int32_t
in_stride_dim2_
;
int32_t
in_stride_dim2_
;
int32_t
out_stride_dim0_
;
int32_t
out_stride_dim0_
;
int32_t
out_stride_dim1_
;
int32_t
out_stride_dim1_
;
int32_t
out_stride_dim2_
;
int32_t
out_stride_dim2_
;
uint8_t
data_type_size_
;
uint8_t
data_type_size_
;
};
}
DepthToSpaceParameter
;
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_PARAMETER_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_DEPTH_TO_SPACE_PARAMETER_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/errorcode.h
浏览文件 @
5405aa5b
...
@@ -17,15 +17,15 @@
...
@@ -17,15 +17,15 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ERRORCODE_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ERRORCODE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ERRORCODE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ERRORCODE_H_
enum
ErrorCodeCommonEnum
{
typedef
enum
ErrorCodeCommonEnum
{
NNACL_OK
=
0
,
NNACL_OK
=
0
,
NNACL_ERR
=
1
,
NNACL_ERR
=
1
,
NNACL_NULL_PTR
,
NNACL_NULL_PTR
,
NNACL_PARAM_INVALID
,
NNACL_PARAM_INVALID
,
OPLIB_COMMON_END
=
9999
OPLIB_COMMON_END
=
9999
};
}
ErrorCodeCommonEnum
;
enum
ErrorCodeFp32OpEnum
{
typedef
enum
ErrorCodeFp32OpEnum
{
NNACL_ERRCODE_OP_FP32_START
=
10000
,
NNACL_ERRCODE_OP_FP32_START
=
10000
,
NNACL_ERRCODE_STRASSEN_RECURSION_MALLOC
,
NNACL_ERRCODE_STRASSEN_RECURSION_MALLOC
,
NNACL_ERRCODE_REVERSE_MALLOC
,
NNACL_ERRCODE_REVERSE_MALLOC
,
...
@@ -35,13 +35,21 @@ enum ErrorCodeFp32OpEnum {
...
@@ -35,13 +35,21 @@ enum ErrorCodeFp32OpEnum {
NNACL_ERRCODE_DIVISOR_ZERO
,
NNACL_ERRCODE_DIVISOR_ZERO
,
NNACL_ERRCODE_INDEX_OUT_OF_RANGE
,
NNACL_ERRCODE_INDEX_OUT_OF_RANGE
,
NNACL_ERRCODE_OP_FP32_END
=
19999
NNACL_ERRCODE_OP_FP32_END
=
19999
};
}
ErrorCodeFp32OpEnum
;
enum
ErrorCodeFp16OpEnum
{
NNACL_ERRCODE_OP_FP16_START
=
20000
,
NNACL_ERRCODE_OP_FP16_END
=
29999
};
typedef
enum
ErrorCodeFp16OpEnum
{
NNACL_ERRCODE_OP_FP16_START
=
20000
,
NNACL_ERRCODE_OP_FP16_END
=
29999
}
ErrorCodeFp16OpEnum
;
enum
ErrorCodeUint8OpEnum
{
NNACL_ERRCODE_OP_UINT8_START
=
30000
,
NNACL_ERRCODE_OP_UINT8_END
=
39999
};
typedef
enum
ErrorCodeUint8OpEnum
{
NNACL_ERRCODE_OP_UINT8_START
=
30000
,
NNACL_ERRCODE_OP_UINT8_END
=
39999
}
ErrorCodeUint8OpEnum
;
enum
ErrorCodeInt8OpEnum
{
NNACL_ERRCODE_OP_INT8_START
=
40000
,
NNACL_ERRCODE_OP_INT8_END
=
49999
};
typedef
enum
ErrorCodeInt8OpEnum
{
NNACL_ERRCODE_OP_INT8_START
=
40000
,
NNACL_ERRCODE_OP_INT8_END
=
49999
}
ErrorCodeInt8OpEnums
;
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ERRORCODE_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ERRORCODE_H_
mindspore/lite/src/runtime/kernel/arm/nnacl/flatten.cc
浏览文件 @
5405aa5b
...
@@ -13,7 +13,7 @@
...
@@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* See the License for the specific language governing permissions and
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/flatten.h"
#include "nnacl/flatten.h"
#include <string.h>
#include <string.h>
void
Flatten
(
const
void
*
input
,
void
*
output
,
FlattenParameter
*
flatten_param
)
{
void
Flatten
(
const
void
*
input
,
void
*
output
,
FlattenParameter
*
flatten_param
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/flatten.h
浏览文件 @
5405aa5b
...
@@ -15,12 +15,12 @@
...
@@ -15,12 +15,12 @@
*/
*/
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FLATTEN_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FLATTEN_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FLATTEN_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FLATTEN_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
FlattenParameter
{
typedef
struct
FlattenParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
size
;
int
size
;
};
}
FlattenParameter
;
void
Flatten
(
const
void
*
input
,
void
*
output
,
FlattenParameter
*
flatten_param
);
void
Flatten
(
const
void
*
input
,
void
*
output
,
FlattenParameter
*
flatten_param
);
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FLATTEN_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FLATTEN_H_
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/common_func.h
浏览文件 @
5405aa5b
...
@@ -20,8 +20,8 @@
...
@@ -20,8 +20,8 @@
#include <stdint.h>
#include <stdint.h>
#include <stdio.h>
#include <stdio.h>
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
#ifdef __cplusplus
#ifdef __cplusplus
extern
"C"
{
extern
"C"
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_depthwise_fp16.cc
浏览文件 @
5405aa5b
...
@@ -14,9 +14,9 @@
...
@@ -14,9 +14,9 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp16/conv_depthwise_fp16.h"
#include "nnacl/fp16/conv_depthwise_fp16.h"
#include <arm_neon.h>
#include <arm_neon.h>
#include "
src/runtime/kernel/arm/
nnacl/fp16/common_func.h"
#include "nnacl/fp16/common_func.h"
/*conv depthwise fp16 begin*/
/*conv depthwise fp16 begin*/
void
DepthwiseBorderPixelFp16
(
float16_t
*
dst
,
const
float16_t
*
src
,
const
float16_t
*
weight
,
const
float16_t
*
bias
,
void
DepthwiseBorderPixelFp16
(
float16_t
*
dst
,
const
float16_t
*
src
,
const
float16_t
*
weight
,
const
float16_t
*
bias
,
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_depthwise_fp16.h
浏览文件 @
5405aa5b
...
@@ -17,8 +17,8 @@
...
@@ -17,8 +17,8 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP16_CONV_DEPTHWISE_FP16_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP16_CONV_DEPTHWISE_FP16_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP16_CONV_DEPTHWISE_FP16_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP16_CONV_DEPTHWISE_FP16_H_
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
#include "
src/runtime/kernel/arm/
nnacl/fp32/conv_depthwise.h"
#include "nnacl/fp32/conv_depthwise.h"
void
ConvDwC8Fp16
(
float16_t
*
output_data
,
const
float16_t
*
input_data
,
const
float16_t
*
weight_data
,
void
ConvDwC8Fp16
(
float16_t
*
output_data
,
const
float16_t
*
input_data
,
const
float16_t
*
weight_data
,
const
float16_t
*
bias_data
,
const
ConvParameter
*
conv_param
,
const
SlidingWindowParam
*
sliding
,
const
float16_t
*
bias_data
,
const
ConvParameter
*
conv_param
,
const
SlidingWindowParam
*
sliding
,
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_fp16.cc
浏览文件 @
5405aa5b
...
@@ -13,10 +13,10 @@
...
@@ -13,10 +13,10 @@
* See the License for the specific language governing permissions and
* See the License for the specific language governing permissions and
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp16/conv_fp16.h"
#include "nnacl/fp16/conv_fp16.h"
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/fp16/pack_fp16.h"
#include "nnacl/fp16/pack_fp16.h"
#include "
src/runtime/kernel/arm/
nnacl/fp16/winograd_transform_fp16.h"
#include "nnacl/fp16/winograd_transform_fp16.h"
extern
"C"
{
extern
"C"
{
#ifdef ENABLE_ARM64
#ifdef ENABLE_ARM64
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/conv_fp16.h
浏览文件 @
5405aa5b
...
@@ -17,7 +17,7 @@
...
@@ -17,7 +17,7 @@
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP16_CONV_FP16_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP16_CONV_FP16_H_
#include <arm_neon.h>
#include <arm_neon.h>
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
#ifndef ENABLE_NEON
#ifndef ENABLE_NEON
void
IndirectGemmFp16_16x8
(
float16_t
*
output
,
float16_t
*
input
,
float16_t
*
weight
,
float16_t
*
bias
,
size_t
step
,
void
IndirectGemmFp16_16x8
(
float16_t
*
output
,
float16_t
*
input
,
float16_t
*
weight
,
float16_t
*
bias
,
size_t
step
,
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/pack_fp16.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp16/pack_fp16.h"
#include "nnacl/fp16/pack_fp16.h"
#include <cstring>
#include <cstring>
#include <cstdlib>
#include <cstdlib>
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/pack_fp16.h
浏览文件 @
5405aa5b
...
@@ -20,8 +20,8 @@
...
@@ -20,8 +20,8 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
void
Im2ColPackUnitFp16
(
float16_t
*
input_data
,
ConvParameter
*
conv_param
,
float16_t
*
packed_input
,
int
real_cal_num
,
void
Im2ColPackUnitFp16
(
float16_t
*
input_data
,
ConvParameter
*
conv_param
,
float16_t
*
packed_input
,
int
real_cal_num
,
int
block_index
);
int
block_index
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/winograd_transform_fp16.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp16/winograd_transform_fp16.h"
#include "nnacl/fp16/winograd_transform_fp16.h"
// for fp16 convolution 3x3 filter/input/output transform F(4,3)
// for fp16 convolution 3x3 filter/input/output transform F(4,3)
void
Conv3x3Fp16InputUnit
(
float16_t
*
tmp_data
,
float16_t
*
trans_input_data
,
size_t
step
)
{
void
Conv3x3Fp16InputUnit
(
float16_t
*
tmp_data
,
float16_t
*
trans_input_data
,
size_t
step
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp16/winograd_transform_fp16.h
浏览文件 @
5405aa5b
...
@@ -19,8 +19,8 @@
...
@@ -19,8 +19,8 @@
#include <arm_neon.h>
#include <arm_neon.h>
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/fp16/pack_fp16.h"
#include "nnacl/fp16/pack_fp16.h"
#include "
src/runtime/kernel/arm/
nnacl/fp16/conv_fp16.h"
#include "nnacl/fp16/conv_fp16.h"
// for fp16 convolution 3x3 filter/input/output transform
// for fp16 convolution 3x3 filter/input/output transform
void
Conv3x3Fp16InputUnit
(
float16_t
*
tmp_data
,
float16_t
*
trans_input_data
,
size_t
step
);
void
Conv3x3Fp16InputUnit
(
float16_t
*
tmp_data
,
float16_t
*
trans_input_data
,
size_t
step
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/activation.h
浏览文件 @
5405aa5b
...
@@ -17,15 +17,15 @@
...
@@ -17,15 +17,15 @@
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ACTIVATION_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_ACTIVATION_H_
#include <math.h>
#include <math.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
#include "
src/runtime/kernel/arm/
nnacl/quantization/fixed_point.h"
#include "nnacl/quantization/fixed_point.h"
struct
ActivationParameter
{
typedef
struct
ActivationParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
type_
;
int
type_
;
float
alpha_
{
0
.
2
};
float
alpha_
{
0
.
2
};
};
}
ActivationParameter
;
inline
int
Relu
(
const
float
*
src
,
int
length
,
float
*
dst
)
{
inline
int
Relu
(
const
float
*
src
,
int
length
,
float
*
dst
)
{
for
(
int
i
=
0
;
i
<
length
;
++
i
)
{
for
(
int
i
=
0
;
i
<
length
;
++
i
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.cc
浏览文件 @
5405aa5b
...
@@ -13,7 +13,7 @@
...
@@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* See the License for the specific language governing permissions and
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/arg_min_max.h"
#include "nnacl/fp32/arg_min_max.h"
#include <stdlib.h>
#include <stdlib.h>
#include <float.h>
#include <float.h>
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.h
浏览文件 @
5405aa5b
...
@@ -16,7 +16,7 @@
...
@@ -16,7 +16,7 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_ARG_MIN_MAX_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_ARG_MIN_MAX_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_ARG_MIN_MAX_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_ARG_MIN_MAX_H_
#include "
src/runtime/kernel/arm/
nnacl/arg_min_max_parameter.h"
#include "nnacl/arg_min_max_parameter.h"
void
ArgMax
(
const
float
*
input
,
float
*
output
,
ArgMinMaxParameter
*
param
,
int
pre_axis_count
,
int
axis_count
,
void
ArgMax
(
const
float
*
input
,
float
*
output
,
ArgMinMaxParameter
*
param
,
int
pre_axis_count
,
int
axis_count
,
int
after_axis_count
);
int
after_axis_count
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,8 @@
...
@@ -14,7 +14,8 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "src/runtime/kernel/arm/nnacl/fp32/arithmetic.h"
#include "nnacl/fp32/arithmetic.h"
#include <math.h>
int
ElementMul
(
float
*
input0
,
float
*
input1
,
float
*
output
,
int
element_size
)
{
int
ElementMul
(
float
*
input0
,
float
*
input1
,
float
*
output
,
int
element_size
)
{
int
block_mod
=
element_size
%
C4NUM
;
int
block_mod
=
element_size
%
C4NUM
;
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic.h
浏览文件 @
5405aa5b
...
@@ -19,9 +19,9 @@
...
@@ -19,9 +19,9 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/arithmetic_common.h"
#include "nnacl/arithmetic_common.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
int
ElementMul
(
float
*
input0
,
float
*
input1
,
float
*
output
,
int
element_size
);
int
ElementMul
(
float
*
input0
,
float
*
input1
,
float
*
output
,
int
element_size
);
int
ElementMulRelu
(
float
*
input0
,
float
*
input1
,
float
*
output
,
int
element_size
);
int
ElementMulRelu
(
float
*
input0
,
float
*
input1
,
float
*
output
,
int
element_size
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic_self.cc
浏览文件 @
5405aa5b
...
@@ -16,7 +16,7 @@
...
@@ -16,7 +16,7 @@
#include <string.h>
#include <string.h>
#include <math.h>
#include <math.h>
#include "
src/runtime/kernel/arm/
nnacl/fp32/arithmetic_self.h"
#include "nnacl/fp32/arithmetic_self.h"
// abs:
// abs:
int
ElementAbs
(
float
*
input
,
float
*
output
,
int
element_size
)
{
int
ElementAbs
(
float
*
input
,
float
*
output
,
int
element_size
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arithmetic_self.h
浏览文件 @
5405aa5b
...
@@ -20,8 +20,8 @@
...
@@ -20,8 +20,8 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
int
ElementAbs
(
float
*
input
,
float
*
output
,
int
element_size
);
int
ElementAbs
(
float
*
input
,
float
*
output
,
int
element_size
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/batchnorm.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,8 @@
...
@@ -14,7 +14,8 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "src/runtime/kernel/arm/nnacl/fp32/batchnorm.h"
#include "nnacl/fp32/batchnorm.h"
#include <math.h>
void
BatchNorm
(
float
*
output_ptr
,
const
float
*
input_ptr
,
const
float
*
mean_ptr
,
const
float
*
variance_ptr
,
int
task_id
,
void
BatchNorm
(
float
*
output_ptr
,
const
float
*
input_ptr
,
const
float
*
mean_ptr
,
const
float
*
variance_ptr
,
int
task_id
,
BatchNormParameter
*
param
)
{
BatchNormParameter
*
param
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/batchnorm.h
浏览文件 @
5405aa5b
...
@@ -17,14 +17,14 @@
...
@@ -17,14 +17,14 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_BATCHNORM_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_BATCHNORM_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_BATCHNORM_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_BATCHNORM_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
BatchNormParameter
{
typedef
struct
BatchNormParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
float
epsilon_
;
float
epsilon_
;
int
unit_
;
int
unit_
;
int
channel_
;
int
channel_
;
};
}
BatchNormParameter
;
void
BatchNorm
(
float
*
output_ptr
,
const
float
*
input_ptr
,
const
float
*
mean_ptr
,
const
float
*
variance_ptr
,
int
task_id
,
void
BatchNorm
(
float
*
output_ptr
,
const
float
*
input_ptr
,
const
float
*
mean_ptr
,
const
float
*
variance_ptr
,
int
task_id
,
BatchNormParameter
*
param
);
BatchNormParameter
*
param
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/broadcast_to.cc
浏览文件 @
5405aa5b
...
@@ -14,9 +14,9 @@
...
@@ -14,9 +14,9 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/broadcast_to.h"
#include "nnacl/fp32/broadcast_to.h"
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
void
PadBroadcastShapeInfo
(
BroadcastShapeInfo
*
shape_info
)
{
void
PadBroadcastShapeInfo
(
BroadcastShapeInfo
*
shape_info
)
{
if
(
shape_info
->
input_shape_size_
<
DIMENSION_4D
)
{
if
(
shape_info
->
input_shape_size_
<
DIMENSION_4D
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/broadcast_to.h
浏览文件 @
5405aa5b
...
@@ -19,22 +19,22 @@
...
@@ -19,22 +19,22 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#define BROADCAST_TO_SHAPE_MAX_SIZE 4
#define BROADCAST_TO_SHAPE_MAX_SIZE 4
struct
BroadcastToParameter
{
typedef
struct
BroadcastToParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
shape_
[
BROADCAST_TO_SHAPE_MAX_SIZE
];
int
shape_
[
BROADCAST_TO_SHAPE_MAX_SIZE
];
size_t
shape_size_
;
size_t
shape_size_
;
};
}
BroadcastToParameter
;
struct
BroadcastShapeInfo
{
typedef
struct
BroadcastShapeInfo
{
int
input_shape_
[
BROADCAST_TO_SHAPE_MAX_SIZE
];
int
input_shape_
[
BROADCAST_TO_SHAPE_MAX_SIZE
];
int
input_shape_size_
;
int
input_shape_size_
;
int
output_shape_
[
BROADCAST_TO_SHAPE_MAX_SIZE
];
int
output_shape_
[
BROADCAST_TO_SHAPE_MAX_SIZE
];
int
output_shape_size_
;
int
output_shape_size_
;
};
}
BroadcastShapeInfo
;
int
BroadcastTo
(
const
float
*
input
,
BroadcastShapeInfo
*
shape_info
,
float
*
output
);
int
BroadcastTo
(
const
float
*
input
,
BroadcastShapeInfo
*
shape_info
,
float
*
output
);
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_BROADCAST_TO_H_
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_BROADCAST_TO_H_
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/cast.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/cast.h"
#include "nnacl/fp32/cast.h"
void
Uint8ToFloat32
(
const
uint8_t
*
input
,
float
*
output
,
int
number
)
{
void
Uint8ToFloat32
(
const
uint8_t
*
input
,
float
*
output
,
int
number
)
{
for
(
int
i
=
0
;
i
<
number
;
++
i
)
{
for
(
int
i
=
0
;
i
<
number
;
++
i
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/cast.h
浏览文件 @
5405aa5b
...
@@ -19,14 +19,14 @@
...
@@ -19,14 +19,14 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
// For cast.
// For cast.
struct
CastParameter
{
typedef
struct
CastParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
src_type_
;
int
src_type_
;
int
dst_type_
;
int
dst_type_
;
};
}
CastParameter
;
void
Uint8ToFloat32
(
const
uint8_t
*
input
,
float
*
output
,
int
number
);
void
Uint8ToFloat32
(
const
uint8_t
*
input
,
float
*
output
,
int
number
);
void
Uint8ToInt8
(
const
uint8_t
*
input
,
int8_t
*
output
,
int
number
);
void
Uint8ToInt8
(
const
uint8_t
*
input
,
int8_t
*
output
,
int
number
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/common_func.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/common_func.h"
#include "nnacl/fp32/common_func.h"
#ifndef __aarch64__
#ifndef __aarch64__
void
MatrixAdd
(
const
float
*
a_ptr
,
const
float
*
b_ptr
,
float
*
dst
,
size_t
a_stride
,
size_t
b_stride
,
size_t
c_stride
,
void
MatrixAdd
(
const
float
*
a_ptr
,
const
float
*
b_ptr
,
float
*
dst
,
size_t
a_stride
,
size_t
b_stride
,
size_t
c_stride
,
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/common_func.h
浏览文件 @
5405aa5b
...
@@ -20,8 +20,8 @@
...
@@ -20,8 +20,8 @@
#include <stdint.h>
#include <stdint.h>
#include <stdio.h>
#include <stdio.h>
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
#ifdef __cplusplus
#ifdef __cplusplus
extern
"C"
{
extern
"C"
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/concat.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/concat.h"
#include "nnacl/fp32/concat.h"
#include <string.h>
#include <string.h>
void
Concat
(
void
**
input
,
int
input_num
,
int
axis
,
int
**
inputs_output_shape
,
size_t
shape_size
,
void
*
output
)
{
void
Concat
(
void
**
input
,
int
input_num
,
int
axis
,
int
**
inputs_output_shape
,
size_t
shape_size
,
void
*
output
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/concat.h
浏览文件 @
5405aa5b
...
@@ -17,7 +17,7 @@
...
@@ -17,7 +17,7 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CONCAT_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CONCAT_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CONCAT_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CONCAT_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
void
Concat
(
void
**
input
,
int
input_num
,
int
axis
,
int
**
inputs_output_shape
,
size_t
shape_size
,
void
*
output
);
void
Concat
(
void
**
input
,
int
input_num
,
int
axis
,
int
**
inputs_output_shape
,
size_t
shape_size
,
void
*
output
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv.cc
浏览文件 @
5405aa5b
...
@@ -14,9 +14,9 @@
...
@@ -14,9 +14,9 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/conv.h"
#include "nnacl/fp32/conv.h"
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/winograd_transform.h"
#include "nnacl/winograd_transform.h"
// fp32 conv common
// fp32 conv common
void
ConvFp32
(
float
*
input_data
,
float
*
packed_input
,
float
*
packed_weight
,
const
float
*
bias_data
,
void
ConvFp32
(
float
*
input_data
,
float
*
packed_input
,
float
*
packed_weight
,
const
float
*
bias_data
,
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv.h
浏览文件 @
5405aa5b
...
@@ -20,12 +20,12 @@
...
@@ -20,12 +20,12 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/pack.h"
#include "nnacl/pack.h"
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/common_func.h"
#include "nnacl/common_func.h"
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
#include "
src/runtime/kernel/arm/
nnacl/fp32/strassen_matmul.h"
#include "nnacl/fp32/strassen_matmul.h"
#include "
src/runtime/kernel/arm/
nnacl/winograd_utils.h"
#include "nnacl/winograd_utils.h"
using
TmpBufferAddress
=
float
*
;
using
TmpBufferAddress
=
float
*
;
typedef
void
(
*
GEMM_FUNC_FP32
)(
float
*
output
,
const
float
*
input
,
const
float
*
weight
,
const
float
*
bias
,
size_t
step
,
typedef
void
(
*
GEMM_FUNC_FP32
)(
float
*
output
,
const
float
*
input
,
const
float
*
weight
,
const
float
*
bias
,
size_t
step
,
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv_depthwise.cc
浏览文件 @
5405aa5b
...
@@ -14,9 +14,9 @@
...
@@ -14,9 +14,9 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/conv_depthwise.h"
#include "nnacl/fp32/conv_depthwise.h"
#include "
src/runtime/kernel/arm/
nnacl/fp32/common_func.h"
#include "nnacl/fp32/common_func.h"
#include "
src/runtime/kernel/arm/
nnacl/winograd_transform.h"
#include "nnacl/winograd_transform.h"
#ifdef ENABLE_ARM64
#ifdef ENABLE_ARM64
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/conv_depthwise.h
浏览文件 @
5405aa5b
...
@@ -17,9 +17,9 @@
...
@@ -17,9 +17,9 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CONV_DEPTHWISE_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CONV_DEPTHWISE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CONV_DEPTHWISE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CONV_DEPTHWISE_H_
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
struct
SlidingWindowParam
{
typedef
struct
SlidingWindowParam
{
int
left_
;
int
left_
;
int
right_
;
int
right_
;
int
top_
;
int
top_
;
...
@@ -35,7 +35,7 @@ struct SlidingWindowParam {
...
@@ -35,7 +35,7 @@ struct SlidingWindowParam {
int
in_kh_step_
;
// kernel H
int
in_kh_step_
;
// kernel H
int
in_kw_step_
;
// kernel W
int
in_kw_step_
;
// kernel W
int
kernel_step_
;
int
kernel_step_
;
};
}
SlidingWindowParam
;
void
InitSlidingParam
(
SlidingWindowParam
*
sliding
,
const
ConvParameter
*
conv_param
,
int
block
);
void
InitSlidingParam
(
SlidingWindowParam
*
sliding
,
const
ConvParameter
*
conv_param
,
int
block
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/crop.cc
浏览文件 @
5405aa5b
...
@@ -13,10 +13,10 @@
...
@@ -13,10 +13,10 @@
* See the License for the specific language governing permissions and
* See the License for the specific language governing permissions and
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/crop.h"
#include "nnacl/fp32/crop.h"
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/crop_parameter.h"
#include "nnacl/crop_parameter.h"
void
Pad4DOffset
(
CropParameter
*
crop_param
,
int64_t
*
offset
)
{
void
Pad4DOffset
(
CropParameter
*
crop_param
,
int64_t
*
offset
)
{
int
axis
=
crop_param
->
axis_
;
int
axis
=
crop_param
->
axis_
;
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/crop.h
浏览文件 @
5405aa5b
...
@@ -15,8 +15,8 @@
...
@@ -15,8 +15,8 @@
*/
*/
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CROP_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CROP_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CROP_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_CROP_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/crop_parameter.h"
#include "nnacl/crop_parameter.h"
#define CROP_OFFSET_MAX_SIZE 4
#define CROP_OFFSET_MAX_SIZE 4
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/deconv.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/deconv.h"
#include "nnacl/fp32/deconv.h"
void
PackDeConvWeightFp32
(
const
float
*
weight
,
float
*
dst
,
int
input_channel
,
int
output_channel
,
int
plane
)
{
void
PackDeConvWeightFp32
(
const
float
*
weight
,
float
*
dst
,
int
input_channel
,
int
output_channel
,
int
plane
)
{
/* ichwoc(nhwc) -> oc4 * h * w * incUP4 * 4 */
/* ichwoc(nhwc) -> oc4 * h * w * incUP4 * 4 */
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/deconv.h
浏览文件 @
5405aa5b
...
@@ -16,10 +16,10 @@
...
@@ -16,10 +16,10 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_DECONV_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_DECONV_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_DECONV_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_DECONV_H_
#include "
src/runtime/kernel/arm/
nnacl/pack.h"
#include "nnacl/pack.h"
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/conv_parameter.h"
#include "nnacl/conv_parameter.h"
#include "
src/runtime/kernel/arm/
nnacl/fp32/strassen_matmul.h"
#include "nnacl/fp32/strassen_matmul.h"
void
PackDeConvWeightFp32
(
const
float
*
weight
,
float
*
dst
,
int
input_channel
,
int
output_channel
,
int
plane
);
void
PackDeConvWeightFp32
(
const
float
*
weight
,
float
*
dst
,
int
input_channel
,
int
output_channel
,
int
plane
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/elu.cc
浏览文件 @
5405aa5b
...
@@ -14,9 +14,10 @@
...
@@ -14,9 +14,10 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "src/runtime/kernel/arm/nnacl/fp32/elu.h"
#include "nnacl/fp32/elu.h"
#include <math.h>
#include "include/errorcode.h"
#include "include/errorcode.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
#include "mindspore/core/utils/log_adapter.h"
#include "mindspore/core/utils/log_adapter.h"
void
Calculate_Data
(
float
*
input_data
,
float
*
output_data
,
int
num
,
EluParameter
*
parameter
)
{
void
Calculate_Data
(
float
*
input_data
,
float
*
output_data
,
int
num
,
EluParameter
*
parameter
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/elu.h
浏览文件 @
5405aa5b
...
@@ -17,9 +17,9 @@
...
@@ -17,9 +17,9 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_ELU_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_ELU_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_ELU_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_ELU_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
typedef
struct
{
typedef
struct
EluParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
float
alpha_
;
float
alpha_
;
int
thread_num_
;
int
thread_num_
;
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/embedding_lookup.cc
浏览文件 @
5405aa5b
...
@@ -14,9 +14,10 @@
...
@@ -14,9 +14,10 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "src/runtime/kernel/arm/nnacl/fp32/embedding_lookup.h"
#include "nnacl/fp32/embedding_lookup.h"
#include <string.h>
#include "include/errorcode.h"
#include "include/errorcode.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
#include "mindspore/core/utils/log_adapter.h"
#include "mindspore/core/utils/log_adapter.h"
void
l2_regulate
(
float
*
data
,
int
size
,
float
max_norm
)
{
void
l2_regulate
(
float
*
data
,
int
size
,
float
max_norm
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/embedding_lookup.h
浏览文件 @
5405aa5b
...
@@ -17,16 +17,16 @@
...
@@ -17,16 +17,16 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_EMBEDDING_LOOKUP_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_EMBEDDING_LOOKUP_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_EMBEDDING_LOOKUP_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_EMBEDDING_LOOKUP_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
typedef
struct
{
typedef
struct
EmbeddingLookupParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
bool
*
is_regulated_
;
bool
*
is_regulated_
;
float
max_norm_
;
float
max_norm_
;
int
ids_size_
;
int
ids_size_
;
int
layer_size_
;
int
layer_size_
;
int
layer_num_
;
int
layer_num_
;
int
thread_num
;
int
thread_num
;
}
EmbeddingLookupParameter
;
}
EmbeddingLookupParameter
;
int
EmbeddingLookup
(
float
*
input_data
,
int
*
ids
,
float
*
output_data
,
EmbeddingLookupParameter
*
parameter
,
int
task_id
);
int
EmbeddingLookup
(
float
*
input_data
,
int
*
ids
,
float
*
output_data
,
EmbeddingLookupParameter
*
parameter
,
int
task_id
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/expandDims.cc
浏览文件 @
5405aa5b
...
@@ -14,9 +14,9 @@
...
@@ -14,9 +14,9 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/expandDims.h"
#include "nnacl/fp32/expandDims.h"
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
int
ExpandDims
(
float
*
input_ptr
,
float
*
output_ptr
,
size_t
data_size
)
{
int
ExpandDims
(
float
*
input_ptr
,
float
*
output_ptr
,
size_t
data_size
)
{
memcpy
(
output_ptr
,
input_ptr
,
data_size
);
memcpy
(
output_ptr
,
input_ptr
,
data_size
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/expandDims.h
浏览文件 @
5405aa5b
...
@@ -17,12 +17,12 @@
...
@@ -17,12 +17,12 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_EXPANDDIMS_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_EXPANDDIMS_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_EXPANDDIMS_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_EXPANDDIMS_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
ExpandDimsParameter
{
typedef
struct
ExpandDimsParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
dim_
;
int
dim_
;
};
}
ExpandDimsParameter
;
int
ExpandDims
(
float
*
input_ptr
,
float
*
output_ptr
,
size_t
data_size
);
int
ExpandDims
(
float
*
input_ptr
,
float
*
output_ptr
,
size_t
data_size
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/fill.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/fill.h"
#include "nnacl/fp32/fill.h"
int
Fill
(
float
*
output
,
int
size
,
float
data
)
{
int
Fill
(
float
*
output
,
int
size
,
float
data
)
{
for
(
int
i
=
0
;
i
<
size
;
++
i
)
{
for
(
int
i
=
0
;
i
<
size
;
++
i
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/fill.h
浏览文件 @
5405aa5b
...
@@ -19,16 +19,16 @@
...
@@ -19,16 +19,16 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
#define FILL_DIMS_MAX_SIZE 4
#define FILL_DIMS_MAX_SIZE 4
struct
FillParameter
{
typedef
struct
FillParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
dims_
[
FILL_DIMS_MAX_SIZE
];
int
dims_
[
FILL_DIMS_MAX_SIZE
];
int
num_dims_
;
int
num_dims_
;
};
}
FillParameter
;
int
Fill
(
float
*
output
,
int
size
,
float
data
);
int
Fill
(
float
*
output
,
int
size
,
float
data
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gather.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/gather.h"
#include "nnacl/fp32/gather.h"
#include <string.h>
#include <string.h>
inline
int
Stride
(
int
*
shape
,
int
rank
,
int
index
)
{
inline
int
Stride
(
int
*
shape
,
int
rank
,
int
index
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gather.h
浏览文件 @
5405aa5b
...
@@ -17,13 +17,13 @@
...
@@ -17,13 +17,13 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_GATHER_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_GATHER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_GATHER_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_GATHER_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
GatherParameter
{
typedef
struct
GatherParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
axis_
;
int
axis_
;
int
batchDims_
;
int
batchDims_
;
};
}
GatherParameter
;
int
Gather
(
float
*
input
,
int
outer_size
,
int
inner_size
,
int
limit
,
int
*
indices
,
int
indices_element_size
,
int
Gather
(
float
*
input
,
int
outer_size
,
int
inner_size
,
int
limit
,
int
*
indices
,
int
indices_element_size
,
float
*
output
);
float
*
output
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gatherNd.cc
浏览文件 @
5405aa5b
...
@@ -14,9 +14,9 @@
...
@@ -14,9 +14,9 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/gatherNd.h"
#include "nnacl/fp32/gatherNd.h"
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
int
GatherNd
(
float
*
input
,
float
*
output
,
int
*
in_offset
,
int
area
,
int
count
)
{
int
GatherNd
(
float
*
input
,
float
*
output
,
int
*
in_offset
,
int
area
,
int
count
)
{
int
i
=
0
;
int
i
=
0
;
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/gatherNd.h
浏览文件 @
5405aa5b
...
@@ -17,12 +17,12 @@
...
@@ -17,12 +17,12 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_GATHERND_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_GATHERND_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_GATHERND_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_GATHERND_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
GatherNdParameter
{
typedef
struct
GatherNdParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
batchDims_
;
int
batchDims_
;
};
}
GatherNdParameter
;
int
GatherNd
(
float
*
input
,
float
*
output
,
int
*
in_offset
,
int
area
,
int
count
);
int
GatherNd
(
float
*
input
,
float
*
output
,
int
*
in_offset
,
int
area
,
int
count
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/local_response_norm.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,8 @@
...
@@ -14,7 +14,8 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "src/runtime/kernel/arm/nnacl/fp32/local_response_norm.h"
#include "nnacl/fp32/local_response_norm.h"
#include <math.h>
int
LocalResponseNorm
(
float
*
input_ptr
,
int
out_size
,
int
channel
,
float
*
output_ptr
,
int
LocalResponseNorm
(
float
*
input_ptr
,
int
out_size
,
int
channel
,
float
*
output_ptr
,
LocalResponseNormParameter
*
param
)
{
LocalResponseNormParameter
*
param
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/local_response_norm.h
浏览文件 @
5405aa5b
...
@@ -17,15 +17,15 @@
...
@@ -17,15 +17,15 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_LOCAL_RESPONSE_NORM_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_LOCAL_RESPONSE_NORM_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_LOCAL_RESPONSE_NORM_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_LOCAL_RESPONSE_NORM_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
LocalResponseNormParameter
{
typedef
struct
LocalResponseNormParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
depth_radius_
;
int
depth_radius_
;
float
bias_
;
float
bias_
;
float
alpha_
;
float
alpha_
;
float
beta_
;
float
beta_
;
};
}
LocalResponseNormParameter
;
int
LocalResponseNorm
(
float
*
input_ptr
,
int
out_size
,
int
channel
,
float
*
output_ptr
,
int
LocalResponseNorm
(
float
*
input_ptr
,
int
out_size
,
int
channel
,
float
*
output_ptr
,
LocalResponseNormParameter
*
param
);
LocalResponseNormParameter
*
param
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/lstm.cc
浏览文件 @
5405aa5b
...
@@ -14,10 +14,10 @@
...
@@ -14,10 +14,10 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/lstm.h"
#include "nnacl/fp32/lstm.h"
#include <string.h>
#include <string.h>
#include "
src/runtime/kernel/arm/
nnacl/fp32/activation.h"
#include "nnacl/fp32/activation.h"
#include "
src/runtime/kernel/arm/
nnacl/fp32/arithmetic.h"
#include "nnacl/fp32/arithmetic.h"
void
InitGate
(
float
*
gate_buffer
,
const
float
*
bias
,
LstmParameter
*
lstm_parm
)
{
void
InitGate
(
float
*
gate_buffer
,
const
float
*
bias
,
LstmParameter
*
lstm_parm
)
{
int
gate_offest
=
0
;
int
gate_offest
=
0
;
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/lstm.h
浏览文件 @
5405aa5b
...
@@ -17,9 +17,9 @@
...
@@ -17,9 +17,9 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_LSTM_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_LSTM_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_LSTM_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_LSTM_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
LstmParameter
{
typedef
struct
LstmParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
input_size_
;
int
input_size_
;
int
hidden_size_
;
// output_size
int
hidden_size_
;
// output_size
...
@@ -28,7 +28,7 @@ struct LstmParameter {
...
@@ -28,7 +28,7 @@ struct LstmParameter {
int
input_step_
;
int
input_step_
;
int
output_step_
;
int
output_step_
;
bool
bidirectional_
;
bool
bidirectional_
;
};
}
LstmParameter
;
void
Lstm
(
float
*
output
,
const
float
*
input
,
const
float
*
weight_i
,
const
float
*
weight_h
,
const
float
*
bias
,
void
Lstm
(
float
*
output
,
const
float
*
input
,
const
float
*
weight_i
,
const
float
*
weight_h
,
const
float
*
bias
,
float
*
hidden_state
,
float
*
cell_state
,
float
*
gate_buffer
,
LstmParameter
*
lstm_parm
);
float
*
hidden_state
,
float
*
cell_state
,
float
*
gate_buffer
,
LstmParameter
*
lstm_parm
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/matmul.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/matmul.h"
#include "nnacl/fp32/matmul.h"
void
RowMajor2Row8Major
(
float
*
src_ptr
,
float
*
dst_ptr
,
int
row
,
int
col
)
{
void
RowMajor2Row8Major
(
float
*
src_ptr
,
float
*
dst_ptr
,
int
row
,
int
col
)
{
for
(
int
r
=
0
;
r
<
row
;
r
++
)
{
for
(
int
r
=
0
;
r
<
row
;
r
++
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/matmul.h
浏览文件 @
5405aa5b
...
@@ -18,9 +18,9 @@
...
@@ -18,9 +18,9 @@
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_MATMUL_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_FP32_MATMUL_H_
#include <float.h>
#include <float.h>
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/nnacl/matmul
.h"
#include "
nnacl/matmul_parameter
.h"
void
MatMul
(
const
float
*
a
,
const
float
*
b
,
float
*
c
,
const
float
*
bias
,
ActType
act_type
,
int
depth
,
int
row
,
int
col
);
void
MatMul
(
const
float
*
a
,
const
float
*
b
,
float
*
c
,
const
float
*
bias
,
ActType
act_type
,
int
depth
,
int
row
,
int
col
);
void
RowMajor2Row8Major
(
float
*
src_ptr
,
float
*
dst_ptr
,
int
row
,
int
col
);
void
RowMajor2Row8Major
(
float
*
src_ptr
,
float
*
dst_ptr
,
int
row
,
int
col
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/one_hot.cc
浏览文件 @
5405aa5b
...
@@ -14,8 +14,8 @@
...
@@ -14,8 +14,8 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/one_hot.h"
#include "nnacl/fp32/one_hot.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
int
OneHot
(
const
int
*
indices
,
float
*
output
,
const
OneHotParameter
*
one_hot_param
,
const
int
tid
,
int
OneHot
(
const
int
*
indices
,
float
*
output
,
const
OneHotParameter
*
one_hot_param
,
const
int
tid
,
const
int
thread_num
)
{
const
int
thread_num
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/one_hot.h
浏览文件 @
5405aa5b
...
@@ -20,9 +20,9 @@
...
@@ -20,9 +20,9 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
OneHotParameter
{
typedef
struct
OneHotParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
axis_
;
int
axis_
;
int
depth_
;
int
depth_
;
...
@@ -30,7 +30,7 @@ struct OneHotParameter {
...
@@ -30,7 +30,7 @@ struct OneHotParameter {
float
off_value_
;
float
off_value_
;
int
outer_size_
;
int
outer_size_
;
int
inner_size_
;
int
inner_size_
;
};
}
OneHotParameter
;
int
OneHot
(
const
int
*
indices
,
float
*
output
,
const
OneHotParameter
*
one_hot_param
,
const
int
tid
,
int
OneHot
(
const
int
*
indices
,
float
*
output
,
const
OneHotParameter
*
one_hot_param
,
const
int
tid
,
const
int
thread_num
);
const
int
thread_num
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pad.cc
浏览文件 @
5405aa5b
...
@@ -14,8 +14,8 @@
...
@@ -14,8 +14,8 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/pad.h"
#include "nnacl/fp32/pad.h"
#include "
src/runtime/kernel/arm/
nnacl/common_func.h"
#include "nnacl/common_func.h"
void
Pad
(
const
float
*
input_data
,
float
*
output_data
,
const
int
*
input_shape
,
const
int
*
output_shape
,
void
Pad
(
const
float
*
input_data
,
float
*
output_data
,
const
int
*
input_shape
,
const
int
*
output_shape
,
const
int
*
paddings
,
const
int
tid
,
const
int
thread_num
)
{
const
int
*
paddings
,
const
int
tid
,
const
int
thread_num
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pad.h
浏览文件 @
5405aa5b
...
@@ -21,8 +21,8 @@
...
@@ -21,8 +21,8 @@
#endif
#endif
#include <memory.h>
#include <memory.h>
#include <float.h>
#include <float.h>
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "
src/runtime/kernel/arm/
nnacl/pad_parameter.h"
#include "nnacl/pad_parameter.h"
void
Pad
(
const
float
*
input_data
,
float
*
output_data
,
const
int
*
input_shape
,
const
int
*
output_shape
,
void
Pad
(
const
float
*
input_data
,
float
*
output_data
,
const
int
*
input_shape
,
const
int
*
output_shape
,
const
int
*
paddings
,
const
int
tid
,
const
int
thread_num
);
const
int
*
paddings
,
const
int
tid
,
const
int
thread_num
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pooling.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/pooling.h"
#include "nnacl/fp32/pooling.h"
#include <float.h>
#include <float.h>
void
AvgPooling
(
const
float
*
input_ptr
,
float
*
output_ptr
,
PoolingParameter
*
pooling_param
,
int
task_id
)
{
void
AvgPooling
(
const
float
*
input_ptr
,
float
*
output_ptr
,
PoolingParameter
*
pooling_param
,
int
task_id
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/pooling.h
浏览文件 @
5405aa5b
...
@@ -20,9 +20,10 @@
...
@@ -20,9 +20,10 @@
#ifdef ENABLE_NEON
#ifdef ENABLE_NEON
#include <arm_neon.h>
#include <arm_neon.h>
#endif
#endif
#include "src/runtime/kernel/arm/nnacl/op_base.h"
#include "nnacl/op_base.h"
#include "nnacl/quantization/quantize.h"
struct
PoolingParameter
{
typedef
struct
PoolingParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
QuantArg
**
quant_args_
;
QuantArg
**
quant_args_
;
bool
global_
;
bool
global_
;
...
@@ -47,7 +48,7 @@ struct PoolingParameter {
...
@@ -47,7 +48,7 @@ struct PoolingParameter {
int
stride_w_
;
int
stride_w_
;
int
stride_h_
;
int
stride_h_
;
int
thread_num_
;
int
thread_num_
;
};
}
PoolingParameter
;
void
AvgPooling
(
const
float
*
input_ptr
,
float
*
output_ptr
,
PoolingParameter
*
pooling_param
,
int
task_id
);
void
AvgPooling
(
const
float
*
input_ptr
,
float
*
output_ptr
,
PoolingParameter
*
pooling_param
,
int
task_id
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/range.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/range.h"
#include "nnacl/fp32/range.h"
void
Range
(
float
*
output_ptr
,
int
start
,
int
limit
,
int
delta
)
{
void
Range
(
float
*
output_ptr
,
int
start
,
int
limit
,
int
delta
)
{
size_t
index
=
0
;
size_t
index
=
0
;
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/range.h
浏览文件 @
5405aa5b
...
@@ -17,15 +17,15 @@
...
@@ -17,15 +17,15 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_RANGE_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_RANGE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_RANGE_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_RANGE_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
struct
RangeParameter
{
typedef
struct
RangeParameter
{
OpParameter
op_parameter_
;
OpParameter
op_parameter_
;
int
dType_
;
int
dType_
;
int
start_
;
int
start_
;
int
limit_
;
int
limit_
;
int
delta_
;
int
delta_
;
};
}
RangeParameter
;
void
Range
(
float
*
output_ptr
,
int
start
,
int
limit
,
int
delta
);
void
Range
(
float
*
output_ptr
,
int
start
,
int
limit
,
int
delta
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/rank.cc
浏览文件 @
5405aa5b
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
* limitations under the License.
* limitations under the License.
*/
*/
#include "
src/runtime/kernel/arm/
nnacl/fp32/rank.h"
#include "nnacl/fp32/rank.h"
void
Rank
(
float
*
output
,
int
rank
)
{
void
Rank
(
float
*
output
,
int
rank
)
{
output
[
0
]
=
(
float
)(
rank
);
output
[
0
]
=
(
float
)(
rank
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/rank.h
浏览文件 @
5405aa5b
...
@@ -16,7 +16,7 @@
...
@@ -16,7 +16,7 @@
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_RANK_H_
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_RANK_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_RANK_H_
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_NNACL_RANK_H_
#include "
src/runtime/kernel/arm/
nnacl/op_base.h"
#include "nnacl/op_base.h"
void
Rank
(
float
*
output
,
int
rank
);
void
Rank
(
float
*
output
,
int
rank
);
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reduce.cc
浏览文件 @
5405aa5b
...
@@ -15,8 +15,8 @@
...
@@ -15,8 +15,8 @@
*/
*/
#include <float.h>
#include <float.h>
#include "
src/runtime/kernel/arm/
nnacl/fp32/reduce.h"
#include "nnacl/fp32/reduce.h"
#include "
src/runtime/kernel/arm/
nnacl/errorcode.h"
#include "nnacl/errorcode.h"
int
ReduceMean
(
const
int
outer_size
,
const
int
inner_size
,
const
int
axis_size
,
const
float
*
src_data
,
int
ReduceMean
(
const
int
outer_size
,
const
int
inner_size
,
const
int
axis_size
,
const
float
*
src_data
,
const
int
*
src_shape
,
float
*
dst_data
,
const
int
tid
,
const
int
thread_num
)
{
const
int
*
src_shape
,
float
*
dst_data
,
const
int
tid
,
const
int
thread_num
)
{
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reduce.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reverse.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/reverse.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/slice.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/slice.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/softmax.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/softmax.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_batch.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_batch.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_depth.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/space_to_depth.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/stack.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/stack.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/strassen_matmul.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/strassen_matmul.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/topk.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/topk.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/unsqueeze.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/unsqueeze.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/activation_grad.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/arithmetic_grad.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/batch_norm.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/batch_norm.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/gemm.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pack_ext.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pack_ext.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pooling_grad.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/pooling_grad.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/reduce_grad.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32_grad/softmax_grad.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fused_batchnorm.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/fused_batchnorm.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arg_min_max_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arg_min_max_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_self_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/arithmetic_self_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/batch_to_space_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/batch_to_space_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/common_func.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/concat_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/concat_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_depthwise_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_depthwise_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/conv_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/crop_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/crop_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/deconv.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/deconv.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/depth_to_space_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/depth_to_space_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/hswish_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/hswish_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/matmul.cc
→
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/matmul
_int8
.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/matmul.h
→
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/matmul
_int8
.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/mul_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/mul_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pad.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pad.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pooling_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/pooling_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/prelu_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/prelu_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/quant_dtype_cast.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/quant_dtype_cast.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/relux_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/reshape_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/reshape_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/sigmoid_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/sigmoid_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/softmax_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/softmax_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/split_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/split_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/squeeze_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/squeeze_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/topk_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/topk_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/unsqueeze_int8.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/int8/unsqueeze_int8.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/matmul.h
→
mindspore/lite/src/runtime/kernel/arm/nnacl/matmul
_parameter
.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/mul_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/nnacl_utils.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/op_base.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/optimized_kernel.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/pack.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/pack.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/pad_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/power.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/power.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/prelu.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/prelu.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/prelu_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/prior_box.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/prior_box.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/quantization/quantize.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/quantization/quantize.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/reshape.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/reshape.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/reshape_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/resize.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/resize.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/reverse_sequence.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/reverse_sequence.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/scale.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/scale.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/scatter_nd.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/scatter_nd.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/shape.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/sigmoid_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/softmax_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/sparse_to_dense.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/sparse_to_dense.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/split.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/split.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/split_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/squeeze.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/squeeze.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/squeeze_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/strassen_matmul.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/strided_slice.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/strided_slice.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/tile.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/tile.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/transpose.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/transpose.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/unique.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/unique.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/unsqueeze_parameter.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/unstack.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/unstack.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/where.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/where.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_transform.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_transform.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_utils.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/winograd_utils.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/zeroslike.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/src/runtime/kernel/arm/nnacl/zeroslike.h
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/conv1x1_fp32_tests.cc
浏览文件 @
5405aa5b
此差异已折叠。
点击以展开。
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录