Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
3236ad7c
Mace
项目概览
Xiaomi
/
Mace
通知
106
Star
40
Fork
27
代码
文件
提交
分支
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
3236ad7c
编写于
3月 01, 2018
作者:
刘
刘琦
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'format' into 'master'
Fix comment formatting See merge request !251
上级
123f4938
e368bc58
变更
59
隐藏空白更改
内联
并排
Showing
59 changed file
with
91 addition
and
91 deletion
+91
-91
mace/core/future.h
mace/core/future.h
+1
-1
mace/core/mace.cc
mace/core/mace.cc
+1
-1
mace/core/net.h
mace/core/net.h
+1
-1
mace/core/operator.h
mace/core/operator.h
+1
-1
mace/core/types.cc
mace/core/types.cc
+1
-1
mace/kernels/bias_add.h
mace/kernels/bias_add.h
+3
-3
mace/kernels/buffer_to_image.h
mace/kernels/buffer_to_image.h
+3
-3
mace/kernels/channel_shuffle.h
mace/kernels/channel_shuffle.h
+2
-2
mace/kernels/concat.h
mace/kernels/concat.h
+3
-3
mace/kernels/conv_pool_2d_util.h
mace/kernels/conv_pool_2d_util.h
+2
-2
mace/kernels/fully_connected.h
mace/kernels/fully_connected.h
+2
-2
mace/kernels/global_avg_pooling.h
mace/kernels/global_avg_pooling.h
+2
-2
mace/kernels/matmul.h
mace/kernels/matmul.h
+2
-2
mace/kernels/neon/conv_2d_neon.cc
mace/kernels/neon/conv_2d_neon.cc
+2
-2
mace/kernels/neon/conv_2d_neon_3x3.cc
mace/kernels/neon/conv_2d_neon_3x3.cc
+2
-2
mace/kernels/neon/conv_2d_neon_5x5.cc
mace/kernels/neon/conv_2d_neon_5x5.cc
+3
-3
mace/kernels/neon/depthwise_conv_neon.cc
mace/kernels/neon/depthwise_conv_neon.cc
+2
-2
mace/kernels/opencl/addn.cc
mace/kernels/opencl/addn.cc
+1
-1
mace/kernels/opencl/concat.cc
mace/kernels/opencl/concat.cc
+1
-1
mace/kernels/opencl/fully_connected_opencl.cc
mace/kernels/opencl/fully_connected_opencl.cc
+1
-1
mace/kernels/opencl/helper.h
mace/kernels/opencl/helper.h
+2
-2
mace/kernels/opencl/matmul.cc
mace/kernels/opencl/matmul.cc
+1
-1
mace/kernels/opencl/space_to_batch_opencl.cc
mace/kernels/opencl/space_to_batch_opencl.cc
+3
-3
mace/kernels/pooling.h
mace/kernels/pooling.h
+2
-2
mace/kernels/softmax.h
mace/kernels/softmax.h
+3
-3
mace/ops/activation.cc
mace/ops/activation.cc
+1
-1
mace/ops/addn.cc
mace/ops/addn.cc
+1
-1
mace/ops/batch_norm.h
mace/ops/batch_norm.h
+2
-2
mace/ops/batch_to_space.cc
mace/ops/batch_to_space.cc
+1
-1
mace/ops/bias_add.cc
mace/ops/bias_add.cc
+1
-1
mace/ops/bias_add.h
mace/ops/bias_add.h
+2
-2
mace/ops/concat.h
mace/ops/concat.h
+2
-2
mace/ops/concat_benchmark.cc
mace/ops/concat_benchmark.cc
+1
-1
mace/ops/core_test.cc
mace/ops/core_test.cc
+1
-1
mace/ops/eltwise.cc
mace/ops/eltwise.cc
+1
-1
mace/ops/eltwise.h
mace/ops/eltwise.h
+2
-2
mace/ops/eltwise_benchmark.cc
mace/ops/eltwise_benchmark.cc
+1
-1
mace/ops/folded_batch_norm.cc
mace/ops/folded_batch_norm.cc
+1
-1
mace/ops/folded_batch_norm.h
mace/ops/folded_batch_norm.h
+2
-2
mace/ops/fully_connected.cc
mace/ops/fully_connected.cc
+1
-1
mace/ops/global_avg_pooling.cc
mace/ops/global_avg_pooling.cc
+1
-1
mace/ops/image_to_buffer.cc
mace/ops/image_to_buffer.cc
+1
-1
mace/ops/image_to_buffer.h
mace/ops/image_to_buffer.h
+2
-2
mace/ops/matmul.cc
mace/ops/matmul.cc
+1
-1
mace/ops/ops_test_util.h
mace/ops/ops_test_util.h
+1
-1
mace/ops/pooling.cc
mace/ops/pooling.cc
+1
-1
mace/ops/reshape.cc
mace/ops/reshape.cc
+1
-1
mace/ops/reshape.h
mace/ops/reshape.h
+2
-2
mace/ops/resize_bilinear.cc
mace/ops/resize_bilinear.cc
+1
-1
mace/ops/softmax.cc
mace/ops/softmax.cc
+1
-1
mace/ops/softmax.h
mace/ops/softmax.h
+2
-2
mace/ops/softmax_benchmark.cc
mace/ops/softmax_benchmark.cc
+1
-1
mace/ops/space_to_batch.cc
mace/ops/space_to_batch.cc
+1
-1
mace/ops/winograd_inverse_transform.cc
mace/ops/winograd_inverse_transform.cc
+1
-1
mace/ops/winograd_transform.cc
mace/ops/winograd_transform.cc
+1
-1
mace/utils/tuner.h
mace/utils/tuner.h
+2
-2
mace/utils/tuner_test.cc
mace/utils/tuner_test.cc
+1
-1
mace/utils/utils.h
mace/utils/utils.h
+2
-2
mace/utils/utils_test.cc
mace/utils/utils_test.cc
+1
-1
未找到文件。
mace/core/future.h
浏览文件 @
3236ad7c
...
...
@@ -20,6 +20,6 @@ struct StatsFuture {
};
};
}
//
namespace mace
}
// namespace mace
#endif // MACE_CORE_FUTURE_H_
mace/core/mace.cc
浏览文件 @
3236ad7c
...
...
@@ -598,4 +598,4 @@ bool MaceEngine::Run(const float *input,
}
}
}
//
namespace mace
}
//
namespace mace
mace/core/net.h
浏览文件 @
3236ad7c
...
...
@@ -63,6 +63,6 @@ std::unique_ptr<NetBase> CreateNet(
DeviceType
type
,
const
NetMode
mode
=
NetMode
::
NORMAL
);
}
//
namespace mace
}
// namespace mace
#endif // MACE_CORE_NET_H_
mace/core/operator.h
浏览文件 @
3236ad7c
...
...
@@ -169,6 +169,6 @@ MACE_DECLARE_REGISTRY(OpRegistry,
#define REGISTER_OPERATOR(op_registry, name, ...) \
MACE_REGISTER_CLASS(OpRegistry, op_registry->registry(), name, __VA_ARGS__)
}
//
namespace mace
}
// namespace mace
#endif // MACE_CORE_OPERATOR_H
mace/core/types.cc
浏览文件 @
3236ad7c
...
...
@@ -74,4 +74,4 @@ size_t GetEnumTypeSize(const DataType dt) {
}
}
}
//
namespace mace
}
// namespace mace
mace/kernels/bias_add.h
浏览文件 @
3236ad7c
...
...
@@ -66,7 +66,7 @@ struct BiasAddFunctor<DeviceType::OPENCL, T> {
cl
::
Kernel
kernel_
;
};
}
//
namepsace kernels
}
//
namespace mace
}
// namepsace kernels
}
// namespace mace
#endif //
MACE_KERNELS_BIAS_ADD_H_
#endif // MACE_KERNELS_BIAS_ADD_H_
mace/kernels/buffer_to_image.h
浏览文件 @
3236ad7c
...
...
@@ -39,7 +39,7 @@ struct BufferToImageFunctor<DeviceType::OPENCL, T> : BufferToImageFunctorBase{
StatsFuture
*
future
);
};
}
//
namepsace kernels
}
//
namespace mace
}
// namepsace kernels
}
// namespace mace
#endif //
MACE_KERNELS_BUFFER_TO_IMAGE_H_
#endif // MACE_KERNELS_BUFFER_TO_IMAGE_H_
mace/kernels/channel_shuffle.h
浏览文件 @
3236ad7c
...
...
@@ -43,7 +43,7 @@ class ChannelShuffleFunctor {
const
int
group_
;
};
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
#endif // MACE_KERNELS_CHANNEL_SHUFFLE_H_
mace/kernels/concat.h
浏览文件 @
3236ad7c
...
...
@@ -85,7 +85,7 @@ struct ConcatFunctor<DeviceType::OPENCL, T> : ConcatFunctorBase{
};
}
//
namepsace kernels
}
//
namespace mace
}
// namepsace kernels
}
// namespace mace
#endif //
MACE_KERNELS_CONCAT_H_
#endif // MACE_KERNELS_CONCAT_H_
mace/kernels/conv_pool_2d_util.h
浏览文件 @
3236ad7c
...
...
@@ -50,7 +50,7 @@ void ConstructNHWCInputWithPadding(const Tensor *input,
Tensor
*
output_tensor
,
bool
padding_same_value
=
false
);
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
#endif // MACE_KERNELS_CONV_POOL_2D_UTIL_H_
mace/kernels/fully_connected.h
浏览文件 @
3236ad7c
...
...
@@ -90,7 +90,7 @@ struct FullyConnectedFunctor<DeviceType::OPENCL, T> : FullyConnectedBase {
cl
::
Kernel
kernel_
;
};
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
#endif // MACE_KERNELS_FULLY_CONNECTED_H_
mace/kernels/global_avg_pooling.h
浏览文件 @
3236ad7c
...
...
@@ -42,7 +42,7 @@ void GlobalAvgPoolingFunctor<DeviceType::NEON, float>::operator()(
const
float
*
input
,
const
index_t
*
input_shape
,
float
*
output
,
StatsFuture
*
future
);
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
#endif // MACE_KERNELS_GLOBAL_AVG_POOLING_H_
mace/kernels/matmul.h
浏览文件 @
3236ad7c
...
...
@@ -63,7 +63,7 @@ struct MatMulFunctor<DeviceType::OPENCL, T> {
cl
::
Kernel
kernel_
;
};
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
#endif // MACE_KERNELS_MATMUL_H_
mace/kernels/neon/conv_2d_neon.cc
浏览文件 @
3236ad7c
...
...
@@ -106,5 +106,5 @@ void Conv2dFunctor<DeviceType::NEON, float>::operator()(const Tensor *input,
bias_data
,
output_data
,
output_shape
);
}
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
mace/kernels/neon/conv_2d_neon_3x3.cc
浏览文件 @
3236ad7c
...
...
@@ -331,5 +331,5 @@ void Conv2dNeonK3x3S2(const float *input, // NCHW
}
}
}
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
mace/kernels/neon/conv_2d_neon_5x5.cc
浏览文件 @
3236ad7c
...
...
@@ -413,7 +413,7 @@ void Conv2dNeonK5x5S1(const float *input, // NCHW
}
}
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
#endif //
MACE_KERNELS_NEON_CONV_2D_NEON_5X5_H_
#endif // MACE_KERNELS_NEON_CONV_2D_NEON_5X5_H_
mace/kernels/neon/depthwise_conv_neon.cc
浏览文件 @
3236ad7c
...
...
@@ -77,5 +77,5 @@ void DepthwiseConv2dFunctor<DeviceType::NEON, float>::operator()(
output_shape
);
}
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
mace/kernels/opencl/addn.cc
浏览文件 @
3236ad7c
...
...
@@ -84,4 +84,4 @@ template
struct
AddNFunctor
<
DeviceType
::
OPENCL
,
half
>;
}
// namespace kernels
}
//
namespace mace
}
// namespace mace
mace/kernels/opencl/concat.cc
浏览文件 @
3236ad7c
...
...
@@ -106,4 +106,4 @@ template
struct
ConcatFunctor
<
DeviceType
::
OPENCL
,
half
>;
}
// namespace kernels
}
//
namespace mace
}
//
namespace mace
mace/kernels/opencl/fully_connected_opencl.cc
浏览文件 @
3236ad7c
...
...
@@ -103,4 +103,4 @@ template
struct
FullyConnectedFunctor
<
DeviceType
::
OPENCL
,
half
>;
}
// namespace kernels
}
//
namespace mace
}
// namespace mace
mace/kernels/opencl/helper.h
浏览文件 @
3236ad7c
...
...
@@ -95,5 +95,5 @@ std::string Concat(Args... args) {
}
}
// namespace kernels
}
//
namespace mace
#endif
//
MACE_KERNELS_OPENCL_HELPER_H_
}
//
namespace mace
#endif
//
MACE_KERNELS_OPENCL_HELPER_H_
mace/kernels/opencl/matmul.cc
浏览文件 @
3236ad7c
...
...
@@ -74,4 +74,4 @@ template
struct
MatMulFunctor
<
DeviceType
::
OPENCL
,
half
>;
}
// namespace kernels
}
//
namespace mace
}
// namespace mace
mace/kernels/opencl/space_to_batch_opencl.cc
浏览文件 @
3236ad7c
...
...
@@ -75,6 +75,6 @@ void SpaceToBatchFunctor<DeviceType::OPENCL, T>::operator()(Tensor *space_tensor
template
struct
SpaceToBatchFunctor
<
DeviceType
::
OPENCL
,
float
>;
template
struct
SpaceToBatchFunctor
<
DeviceType
::
OPENCL
,
half
>;
}
//
namespace kernels
}
//
namespace mace
#endif
//
MACE_KERNELS_OPENCL_SPACE_TO_BATCH_H_
}
//
namespace kernels
}
//
namespace mace
#endif
//
MACE_KERNELS_OPENCL_SPACE_TO_BATCH_H_
mace/kernels/pooling.h
浏览文件 @
3236ad7c
...
...
@@ -175,7 +175,7 @@ struct PoolingFunctor<DeviceType::OPENCL, T> : PoolingFunctorBase {
cl
::
Kernel
kernel_
;
};
}
//
namespace kernels
}
//
namespace mace
}
// namespace kernels
}
// namespace mace
#endif // MACE_KERNELS_POOLING_H
mace/kernels/softmax.h
浏览文件 @
3236ad7c
...
...
@@ -59,7 +59,7 @@ struct SoftmaxFunctor<DeviceType::OPENCL, T> {
cl
::
Kernel
kernel_
;
};
}
//
namepsace kernels
}
//
namespace mace
}
// namepsace kernels
}
// namespace mace
#endif //
MACE_KERNELS_SOFTMAX_H_
#endif // MACE_KERNELS_SOFTMAX_H_
mace/ops/activation.cc
浏览文件 @
3236ad7c
...
...
@@ -26,4 +26,4 @@ void Register_Activation(OperatorRegistry *op_registry) {
ActivationOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/addn.cc
浏览文件 @
3236ad7c
...
...
@@ -26,4 +26,4 @@ void Register_AddN(OperatorRegistry *op_registry) {
AddNOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/batch_norm.h
浏览文件 @
3236ad7c
...
...
@@ -55,6 +55,6 @@ class BatchNormOp : public Operator<D, T> {
OP_OUTPUT_TAGS
(
OUTPUT
);
};
}
//
namespace mace
}
// namespace mace
#endif //
MACE_OPS_BATCH_NORM_H_
#endif // MACE_OPS_BATCH_NORM_H_
mace/ops/batch_to_space.cc
浏览文件 @
3236ad7c
...
...
@@ -19,4 +19,4 @@ void Register_BatchToSpaceND(OperatorRegistry *op_registry) {
BatchToSpaceNDOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/bias_add.cc
浏览文件 @
3236ad7c
...
...
@@ -26,4 +26,4 @@ void Register_BiasAdd(OperatorRegistry *op_registry) {
BiasAddOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/bias_add.h
浏览文件 @
3236ad7c
...
...
@@ -40,6 +40,6 @@ class BiasAddOp : public Operator<D, T> {
OP_OUTPUT_TAGS
(
OUTPUT
);
};
}
//
namespace mace
}
// namespace mace
#endif //
MACE_BIAS_ADD_H_
#endif // MACE_BIAS_ADD_H_
mace/ops/concat.h
浏览文件 @
3236ad7c
...
...
@@ -40,6 +40,6 @@ class ConcatOp : public Operator<D, T> {
OP_OUTPUT_TAGS
(
OUTPUT
);
};
}
//
namespace mace
}
// namespace mace
#endif //
MACE_OPS_CONCAT_H_
#endif // MACE_OPS_CONCAT_H_
mace/ops/concat_benchmark.cc
浏览文件 @
3236ad7c
...
...
@@ -101,4 +101,4 @@ static void BM_CONCATOPENCLHalf(int iters, int dim1) {
BENCHMARK
(
BM_CONCATOPENCLFloat
)
->
Arg
(
32
)
->
Arg
(
64
)
->
Arg
(
128
)
->
Arg
(
256
);
BENCHMARK
(
BM_CONCATOPENCLHalf
)
->
Arg
(
32
)
->
Arg
(
64
)
->
Arg
(
128
)
->
Arg
(
256
);
}
// namespace mace
\ No newline at end of file
}
// namespace mace
\ No newline at end of file
mace/ops/core_test.cc
浏览文件 @
3236ad7c
...
...
@@ -56,4 +56,4 @@ TEST(CoreTest, INIT_MODE) {
1e-5
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/eltwise.cc
浏览文件 @
3236ad7c
...
...
@@ -26,4 +26,4 @@ void Register_Eltwise(OperatorRegistry *op_registry) {
EltwiseOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/eltwise.h
浏览文件 @
3236ad7c
...
...
@@ -42,6 +42,6 @@ class EltwiseOp : public Operator<D, T> {
OP_OUTPUT_TAGS
(
OUTPUT
);
};
}
//
namespace mace
}
// namespace mace
#endif //
MACE_OPS_RESHAPE_H_
#endif // MACE_OPS_RESHAPE_H_
mace/ops/eltwise_benchmark.cc
浏览文件 @
3236ad7c
...
...
@@ -81,4 +81,4 @@ BM_ELTWISE(0, 1, 240, 240, 256);
BM_ELTWISE
(
1
,
1
,
240
,
240
,
256
);
BM_ELTWISE
(
2
,
1
,
240
,
240
,
256
);
}
//
namespace mace
}
// namespace mace
mace/ops/folded_batch_norm.cc
浏览文件 @
3236ad7c
...
...
@@ -29,4 +29,4 @@ void Register_FoldedBatchNorm(OperatorRegistry *op_registry) {
FoldedBatchNormOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/folded_batch_norm.h
浏览文件 @
3236ad7c
...
...
@@ -49,6 +49,6 @@ class FoldedBatchNormOp : public Operator<D, T> {
OP_OUTPUT_TAGS
(
OUTPUT
);
};
}
//
namespace mace
}
// namespace mace
#endif //
MACE_OPS_FOLDED_BATCH_NORM_H_
#endif // MACE_OPS_FOLDED_BATCH_NORM_H_
mace/ops/fully_connected.cc
浏览文件 @
3236ad7c
...
...
@@ -26,4 +26,4 @@ void Register_FullyConnected(OperatorRegistry *op_registry) {
FullyConnectedOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/global_avg_pooling.cc
浏览文件 @
3236ad7c
...
...
@@ -14,4 +14,4 @@ void Register_GlobalAvgPooling(OperatorRegistry *op_registry) {
GlobalAvgPoolingOp
<
DeviceType
::
CPU
,
float
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/image_to_buffer.cc
浏览文件 @
3236ad7c
...
...
@@ -20,4 +20,4 @@ void Register_ImageToBuffer(OperatorRegistry *op_registry) {
ImageToBufferOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/image_to_buffer.h
浏览文件 @
3236ad7c
...
...
@@ -34,5 +34,5 @@ class ImageToBufferOp: public Operator<D, T> {
OP_OUTPUT_TAGS
(
OUTPUT
);
};
}
//
namespace mace
#endif
//
MACE_OPS_IMAGE_TO_BUFFER_H_
}
//
namespace mace
#endif
//
MACE_OPS_IMAGE_TO_BUFFER_H_
mace/ops/matmul.cc
浏览文件 @
3236ad7c
...
...
@@ -26,4 +26,4 @@ void Register_MatMul(OperatorRegistry *op_registry) {
MatMulOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/ops_test_util.h
浏览文件 @
3236ad7c
...
...
@@ -405,4 +405,4 @@ void ImageToBuffer(OpsTestNet &net,
}
// namespace mace
#endif //
MACE_OPS_TEST_UTIL_H_
#endif // MACE_OPS_TEST_UTIL_H_
mace/ops/pooling.cc
浏览文件 @
3236ad7c
...
...
@@ -30,4 +30,4 @@ void Register_Pooling(OperatorRegistry *op_registry) {
PoolingOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/reshape.cc
浏览文件 @
3236ad7c
...
...
@@ -14,4 +14,4 @@ void Register_Reshape(OperatorRegistry *op_registry) {
ReshapeOp
<
DeviceType
::
CPU
,
float
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/reshape.h
浏览文件 @
3236ad7c
...
...
@@ -59,6 +59,6 @@ class ReshapeOp : public Operator<D, T> {
OP_OUTPUT_TAGS
(
OUTPUT
);
};
}
//
namespace mace
}
// namespace mace
#endif //
MACE_OPS_RESHAPE_H_
#endif // MACE_OPS_RESHAPE_H_
mace/ops/resize_bilinear.cc
浏览文件 @
3236ad7c
...
...
@@ -26,4 +26,4 @@ void Register_ResizeBilinear(OperatorRegistry *op_registry) {
ResizeBilinearOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/softmax.cc
浏览文件 @
3236ad7c
...
...
@@ -26,4 +26,4 @@ void Register_Softmax(OperatorRegistry *op_registry) {
SoftmaxOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/softmax.h
浏览文件 @
3236ad7c
...
...
@@ -35,6 +35,6 @@ class SoftmaxOp : public Operator<D, T> {
OP_OUTPUT_TAGS
(
OUTPUT
);
};
}
//
namespace mace
}
// namespace mace
#endif //
MACE_SOFTMAX_H_
#endif // MACE_SOFTMAX_H_
mace/ops/softmax_benchmark.cc
浏览文件 @
3236ad7c
...
...
@@ -65,4 +65,4 @@ BM_SOFTMAX(1, 3, 128, 128);
BM_SOFTMAX
(
1
,
3
,
512
,
512
);
BM_SOFTMAX
(
1
,
32
,
112
,
112
);
BM_SOFTMAX
(
1
,
64
,
256
,
256
);
}
//
namespace mace
}
// namespace mace
mace/ops/space_to_batch.cc
浏览文件 @
3236ad7c
...
...
@@ -19,4 +19,4 @@ void Register_SpaceToBatchND(OperatorRegistry *op_registry) {
SpaceToBatchNDOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/winograd_inverse_transform.cc
浏览文件 @
3236ad7c
...
...
@@ -19,4 +19,4 @@ void Register_WinogradInverseTransform(OperatorRegistry *op_registry) {
WinogradInverseTransformOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/ops/winograd_transform.cc
浏览文件 @
3236ad7c
...
...
@@ -19,4 +19,4 @@ void Register_WinogradTransform(OperatorRegistry *op_registry) {
WinogradTransformOp
<
DeviceType
::
OPENCL
,
half
>
);
}
}
//
namespace mace
}
// namespace mace
mace/utils/tuner.h
浏览文件 @
3236ad7c
...
...
@@ -180,5 +180,5 @@ class Tuner {
std
::
unordered_map
<
std
::
string
,
std
::
vector
<
param_type
>>
param_table_
;
};
}
//
namespace mace
#endif //
MACE_UTILS_TUNER_H_
}
// namespace mace
#endif // MACE_UTILS_TUNER_H_
mace/utils/tuner_test.cc
浏览文件 @
3236ad7c
...
...
@@ -81,4 +81,4 @@ TEST_F(TunerTest, SimpleTune) {
EXPECT_EQ
(
expect
,
res
);
}
}
//
namespace mace
}
//
namespace mace
mace/utils/utils.h
浏览文件 @
3236ad7c
...
...
@@ -94,5 +94,5 @@ inline std::string ObfuscateSymbol(const std::string &src) {
#define MACE_OBFUSCATE_SYMBOL(str) (str)
#endif
}
//
namespace mace
#endif //
MACE_UTILS_UTILS_H_
}
// namespace mace
#endif // MACE_UTILS_UTILS_H_
mace/utils/utils_test.cc
浏览文件 @
3236ad7c
...
...
@@ -81,4 +81,4 @@ TEST_F(TunerTest, SimpleTune) {
EXPECT_EQ
(
expect
,
res
);
}
}
//
namespace mace
}
//
namespace mace
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录