Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
慢慢CG
Mace
提交
fd83af42
Mace
项目概览
慢慢CG
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Mace
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
fd83af42
编写于
4月 01, 2019
作者:
B
Bin Li
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update hexagon nn to 2.6
上级
19084971
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
124 addition
and
4 deletion
+124
-4
mace/libmace/mace.cc
mace/libmace/mace.cc
+3
-1
third_party/nnlib/libhexagon_controller.so
third_party/nnlib/libhexagon_controller.so
+0
-0
third_party/nnlib/ops.h
third_party/nnlib/ops.h
+121
-3
third_party/nnlib/v60/libhexagon_nn_skel.so
third_party/nnlib/v60/libhexagon_nn_skel.so
+0
-0
third_party/nnlib/v66/libhexagon_nn_skel.so
third_party/nnlib/v66/libhexagon_nn_skel.so
+0
-0
未找到文件。
mace/libmace/mace.cc
浏览文件 @
fd83af42
...
...
@@ -485,7 +485,9 @@ MaceStatus MaceEngine::Impl::Init(
<<
MakeString
(
MapKeys
(
output_info_map_
));
}
#if defined(MACE_ENABLE_HEXAGON) || defined(MACE_ENABLE_HTA)
ws_
->
CreateTensor
(
output_name
,
device_
->
allocator
(),
DT_FLOAT
);
Tensor
*
output_tensor
=
ws_
->
CreateTensor
(
output_name
,
device_
->
allocator
(),
DT_FLOAT
);
output_tensor
->
set_data_format
(
NHWC
);
#endif
}
#if defined(MACE_ENABLE_HEXAGON) || defined(MACE_ENABLE_HTA)
...
...
third_party/nnlib/libhexagon_controller.so
浏览文件 @
fd83af42
无法预览此类型文件
third_party/nnlib/ops.h
浏览文件 @
fd83af42
...
...
@@ -218,6 +218,7 @@ DEF_OP(Sink)
DEF_OP_WREF
(
QuantizedPRelu_8_d32
)
DEF_OP_WREF
(
AutoQuantize
)
DEF_OP_WREF
(
QuantizedDepthwiseConv2d_8x8to32
)
DEF_OP
(
QuantizedTransposeConv2d_8x8p32to8
)
DEF_OP_WREF
(
DepthwiseConv2d_f
)
DEF_OP
(
DepthwiseSupernode_8x8p8to8
)
DEF_OP
(
DepthwiseSupernode_8x8p8to8_d32
)
...
...
@@ -226,7 +227,7 @@ DEF_OP_WREF(QuantizedMul_8x8to8_d32)
DEF_OP
(
FullyConnected_u8
)
#if 0
DEF_OP_WREF(QuantizedFC_8x8p8to8)
DEF_OP_WREF(QuantizedFC_8x8p8to8)
#endif
DEF_OP_WREF
(
QuantizedAdd_8p8to8_d32
)
...
...
@@ -291,11 +292,128 @@ DEF_OP_WREF(QuantizedInstanceNorm_8_d32)
DEF_OP_WREF
(
QuantizedInstanceNormBG_8
)
DEF_OP_WREF
(
QuantizedInstanceNormBG_8_d32
)
DEF_OP
(
SuperFC_8x8p32to8
)
DEF_OP
(
SuperFC_8x8p32to8_ref
)
DEF_OP
(
SuperFC_8x8p32to8_d32
)
DEF_OP
(
ChannelShuffle_f
)
DEF_OP
(
ChannelShuffle_int32
)
DEF_OP
(
QuantizedChannelShuffle_8
)
DEF_OP_WREF
(
QuantizedChannelShuffle_8
)
DEF_OP
(
QuantizedChannelShuffle_8_d32
)
/* this is in op_chanshuffle_d32.c*/
DEF_OP
(
QuantizedSplit_8_d32
)
DEF_OP
(
QuantizedCrop_8
)
DEF_OP
(
ResizeUnitSquare_f
)
DEF_OP_WREF
(
ResizeUnitSquare_8
)
DEF_OP_WREF
(
Nv21ToRgb_8
)
DEF_OP_WREF
(
RgbaToRgb_8
)
DEF_OP_WREF
(
Argb32ToRgb_8
)
DEF_OP
(
Permute_f
)
DEF_OP
(
QuantizedPermute_8
)
DEF_OP_WREF
(
QuantizedRoiPool_8
)
DEF_OP
(
Proposal_f
)
DEF_OP
(
RoiAlign_f
)
DEF_OP_WREF
(
QuantizedRoiAlign_8
)
DEF_OP_WREF
(
Implode_8
)
DEF_OP
(
QuantizedConcat_8_nond32
)
DEF_OP
(
Close_16tof
)
DEF_OP
(
QuantizedLstmInput_16x16to16
)
DEF_OP
(
QuantizedLstmOutput_16x16to8
)
DEF_OP
(
Quantize_16
)
DEF_OP
(
Dequantize_16
)
DEF_OP
(
Convert_8_16
)
DEF_OP
(
QuantizedTanh_16
)
DEF_OP
(
QuantizedSigmoid_16
)
DEF_OP_WREF
(
QuantizeDownAndShrinkRange_32to16
)
DEF_OP_WREF
(
Requantize_32to16
)
DEF_OP_WREF
(
QuantizedMatMul_8x8p32to16
)
DEF_OP
(
QuantizedStridedSlice_8
)
DEF_OP
(
Bbox_Transform_f
)
DEF_OP
(
Softmax_uint8
)
DEF_OP
(
QuantizedFakeConcat_8_d32
)
DEF_OP
(
DepthToSpace_8_d32
)
DEF_OP
(
OemNode
)
DEF_OP
(
QuantizedPad_8_d32
)
DEF_OP
(
QuantizedSqrt_8
)
DEF_OP
(
QuantizedSlice_16
)
DEF_OP
(
QuantizedMin_8
)
DEF_OP
(
QuantizedMax_8
)
DEF_OP
(
Transpose_8
)
DEF_OP
(
Close_u16tof
)
DEF_OP
(
QuantizeForTest_16b_d32
)
DEF_OP
(
QuantizeForTest_u16b_d32
)
DEF_OP
(
Close_16b_d32
)
DEF_OP
(
Close_u16b_d32
)
DEF_OP
(
Convert_from_d32_b16
)
DEF_OP
(
Supernode_16x16p16to16_d32
)
DEF_OP
(
Supernode_u16x16p16to16_d32
)
DEF_OP
(
QuantizedMatMulDims_8x8p32to16
)
DEF_OP
(
BatchSeqConfig
)
DEF_OP
(
QuantizedDiv_8
)
DEF_OP
(
QuantizedRecip_8
)
DEF_OP
(
QuantizedNeg_8
)
DEF_OP
(
QuantizedNeg_8_d32
)
DEF_OP
(
Neg_int32
)
DEF_OP
(
Abs_f
)
DEF_OP
(
Abs_int32
)
DEF_OP
(
QuantizedSub_8p8to8
)
DEF_OP
(
Box_Decoder_f
)
DEF_OP
(
QuantizedExtractGlimpse_8
)
DEF_OP
(
QuantizedTile_8
)
DEF_OP_WREF
(
QuantizedMul_8x8to8
)
DEF_OP
(
QuantizedSum_8to32
)
DEF_OP
(
ImageTransform_f
)
DEF_OP
(
Convert_to_aix_d32
)
DEF_OP
(
Convert_from_aix
)
DEF_OP
(
Convert_from_aix_d32
)
DEF_OP
(
MultiClassNms_f
)
DEF_OP
(
BatchToSpaceND_8_d32
)
DEF_OP
(
SpaceToBatchND_8_d32
)
DEF_OP
(
Supernode3322_8x8p8to8
)
DEF_OP
(
Supernode3322_8x8p32to8
)
DEF_OP
(
Convert_int32_f
)
DEF_OP
(
ArgMax_ftoInt32
)
DEF_OP
(
ArgMax_8toInt32
)
DEF_OP
(
ArgMax_8
)
DEF_OP
(
Supernode_8x8p32to8_ref
)
DEF_OP
(
HeatmapMaxKP_f
)
DEF_OP
(
TopK_f
)
DEF_OP
(
TopK_8
)
DEF_OP
(
CastFloat32ToInt32
)
DEF_OP
(
CastFloat32ToUInt8
)
DEF_OP
(
CastInt32ToFloat32
)
DEF_OP
(
CastInt32ToUInt8
)
DEF_OP
(
CastUInt8ToFloat32
)
DEF_OP
(
CastUInt8ToInt32
)
DEF_OP
(
AxisShuffle_8
)
DEF_OP
(
ResizeNearestNeighbor_8
)
DEF_OP
(
QuantizedHeatmapMaxKP_8
)
DEF_OP
(
Moments_8to32
)
DEF_OP
(
ArgMin_8
)
DEF_OP
(
Select_f
)
DEF_OP
(
Select_8
)
DEF_OP
(
QuantizedGroupedConv2d_8x8p32to8
)
// Add new operations above this line
#ifdef __SELF_DEF_OP_WREF
#undef __SELF_DEF_OP_WREF
#undef DEF_OP_WREF
...
...
third_party/nnlib/v60/libhexagon_nn_skel.so
浏览文件 @
fd83af42
无法预览此类型文件
third_party/nnlib/v66/libhexagon_nn_skel.so
浏览文件 @
fd83af42
无法预览此类型文件
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录