Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
e8ee9dc7
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看板
未验证
提交
e8ee9dc7
编写于
7月 09, 2018
作者:
Y
yuyang18
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Several Polish
上级
0d2ccfbd
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
9 addition
and
17 deletion
+9
-17
paddle/fluid/framework/reader.cc
paddle/fluid/framework/reader.cc
+1
-1
paddle/fluid/framework/reader.h
paddle/fluid/framework/reader.h
+4
-11
paddle/fluid/operators/reader/create_py_reader_op.cc
paddle/fluid/operators/reader/create_py_reader_op.cc
+2
-2
paddle/fluid/operators/reader/create_recordio_file_reader_op.cc
.../fluid/operators/reader/create_recordio_file_reader_op.cc
+2
-3
未找到文件。
paddle/fluid/framework/reader.cc
浏览文件 @
e8ee9dc7
...
@@ -36,7 +36,7 @@ void ReaderBase::Start() {
...
@@ -36,7 +36,7 @@ void ReaderBase::Start() {
}
}
}
}
ReaderBase
::~
ReaderBase
()
{}
ReaderBase
::~
ReaderBase
()
{
Shutdown
();
}
}
// namespace framework
}
// namespace framework
}
// namespace paddle
}
// namespace paddle
paddle/fluid/framework/reader.h
浏览文件 @
e8ee9dc7
...
@@ -39,9 +39,9 @@ class ReaderBase {
...
@@ -39,9 +39,9 @@ class ReaderBase {
protected:
protected:
virtual
void
ReadNextImpl
(
std
::
vector
<
LoDTensor
>*
out
)
=
0
;
virtual
void
ReadNextImpl
(
std
::
vector
<
LoDTensor
>*
out
)
=
0
;
virtual
void
ShutdownImpl
()
=
0
;
virtual
void
ShutdownImpl
()
{}
virtual
void
StartImpl
()
=
0
;
virtual
void
StartImpl
()
{}
ReaderStatus
status_
{
kRunning
};
ReaderStatus
status_
{
kRunning
};
};
};
...
@@ -61,15 +61,8 @@ class DecoratedReader : public ReaderBase {
...
@@ -61,15 +61,8 @@ class DecoratedReader : public ReaderBase {
std
::
shared_ptr
<
ReaderBase
>
reader_
;
std
::
shared_ptr
<
ReaderBase
>
reader_
;
};
};
class
FileReader
:
public
ReaderBase
{
// FileReader is just a conceptual class.
public:
class
FileReader
:
public
ReaderBase
{};
FileReader
()
:
ReaderBase
()
{}
protected:
void
ShutdownImpl
()
override
{}
void
StartImpl
()
override
{}
};
// The ReaderHolder is used as reader' unified wrapper,
// The ReaderHolder is used as reader' unified wrapper,
// making it easier to access different type reader in Variables.
// making it easier to access different type reader in Variables.
...
...
paddle/fluid/operators/reader/create_py_reader_op.cc
浏览文件 @
e8ee9dc7
...
@@ -56,8 +56,8 @@ class CreatePyReaderOp : public framework::OperatorBase {
...
@@ -56,8 +56,8 @@ class CreatePyReaderOp : public framework::OperatorBase {
const
std
::
string
&
queue_name
=
Input
(
"blocking_queue"
);
const
std
::
string
&
queue_name
=
Input
(
"blocking_queue"
);
auto
*
queue_holder_var
=
scope
.
FindVar
(
queue_name
);
auto
*
queue_holder_var
=
scope
.
FindVar
(
queue_name
);
PADDLE_ENFORCE
(
PADDLE_ENFORCE
_NOT_NULL
(
queue_holder_var
!=
nullptr
,
queue_holder_var
,
"No LoDTensorBlockingQueueHolder variable with name %s found"
,
"No LoDTensorBlockingQueueHolder variable with name %s found"
,
queue_name
);
queue_name
);
auto
*
queue_holder
=
auto
*
queue_holder
=
...
...
paddle/fluid/operators/reader/create_recordio_file_reader_op.cc
浏览文件 @
e8ee9dc7
...
@@ -22,8 +22,7 @@ template <bool ThreadSafe>
...
@@ -22,8 +22,7 @@ template <bool ThreadSafe>
class
RecordIOFileReader
:
public
framework
::
FileReader
{
class
RecordIOFileReader
:
public
framework
::
FileReader
{
public:
public:
explicit
RecordIOFileReader
(
const
std
::
string
&
filename
)
explicit
RecordIOFileReader
(
const
std
::
string
&
filename
)
:
FileReader
(),
:
scanner_
(
filename
),
scanner_
(
filename
),
dev_ctx_
(
*
platform
::
DeviceContextPool
::
Instance
().
Get
(
dev_ctx_
(
*
platform
::
DeviceContextPool
::
Instance
().
Get
(
platform
::
CPUPlace
()))
{
platform
::
CPUPlace
()))
{
if
(
ThreadSafe
)
{
if
(
ThreadSafe
)
{
...
@@ -42,7 +41,7 @@ class RecordIOFileReader : public framework::FileReader {
...
@@ -42,7 +41,7 @@ class RecordIOFileReader : public framework::FileReader {
}
}
}
}
void
S
hutdown
Impl
()
override
{
scanner_
.
Reset
();
}
void
S
tart
Impl
()
override
{
scanner_
.
Reset
();
}
private:
private:
std
::
unique_ptr
<
std
::
mutex
>
mutex_
;
std
::
unique_ptr
<
std
::
mutex
>
mutex_
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录