Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
a5494fa8
P
Paddle
项目概览
机器未来
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
a5494fa8
编写于
10月 31, 2017
作者:
H
hedaoyuan
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Remove SparseRowMatrix in mobile inference.
上级
f122a5da
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
32 addition
and
2 deletion
+32
-2
paddle/math/SparseRowMatrix.h
paddle/math/SparseRowMatrix.h
+26
-0
paddle/parameter/Parameter.cpp
paddle/parameter/Parameter.cpp
+6
-2
未找到文件。
paddle/math/SparseRowMatrix.h
浏览文件 @
a5494fa8
...
...
@@ -14,6 +14,8 @@ limitations under the License. */
#pragma once
#ifndef PADDLE_MOBILE_INFERENCE
#include <gflags/gflags.h>
#include <string.h>
#include <algorithm>
...
...
@@ -313,3 +315,27 @@ private:
};
}
// namespace paddle
#else
namespace
paddle
{
class
SparseRowCpuMatrix
:
public
CpuMatrix
{
public:
void
reserveStore
()
{}
void
clearIndices
()
{}
};
class
SparsePrefetchRowCpuMatrix
:
public
SparseRowCpuMatrix
{
public:
void
setupIndices
()
{}
void
addRows
(
MatrixPtr
input
)
{}
void
addRows
(
IVectorPtr
ids
)
{}
};
class
SparseAutoGrowRowCpuMatrix
:
public
SparseRowCpuMatrix
{};
class
CacheRowCpuMatrix
:
public
SparseAutoGrowRowCpuMatrix
{};
class
SparseRowIdsCpuMatrix
:
public
CpuMatrix
{};
}
// namespace paddle
#endif
paddle/parameter/Parameter.cpp
浏览文件 @
a5494fa8
...
...
@@ -217,7 +217,9 @@ void Parameter::setMat(ParameterType pType, int matType) {
bufs_
[
pType
]
->
getMemoryHandle
()),
height
,
width
);
}
else
if
(
matType
==
MAT_SPARSE_ROW_IDS
)
{
}
#ifndef PADDLE_MOBILE_INFERENCE
else
if
(
matType
==
MAT_SPARSE_ROW_IDS
)
{
CHECK_EQ
(
height
*
width
,
bufs_
[
pType
]
->
getSize
());
mats_
[
pType
]
=
std
::
make_shared
<
SparseRowIdsCpuMatrix
>
(
std
::
dynamic_pointer_cast
<
CpuMemoryHandle
>
(
...
...
@@ -259,7 +261,9 @@ void Parameter::setMat(ParameterType pType, int matType) {
}
else
if
(
matType
==
MAT_SPARSE_ROW_AUTO_GROW
)
{
CHECK
(
isGradSparseUpdate
());
mats_
[
pType
]
=
std
::
make_shared
<
SparseAutoGrowRowCpuMatrix
>
(
height
,
width
);
}
else
{
}
#endif
else
{
LOG
(
FATAL
)
<<
"Unsupported mat type"
<<
matType
;
}
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录