Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
Paddle
提交
602e28bf
P
Paddle
项目概览
BaiXuePrincess
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
602e28bf
编写于
5月 25, 2018
作者:
T
tensor-tang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
use the actual data
上级
1b8b253e
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
23 addition
and
25 deletion
+23
-25
paddle/fluid/inference/tests/book/test_inference_nlp.cc
paddle/fluid/inference/tests/book/test_inference_nlp.cc
+23
-25
未找到文件。
paddle/fluid/inference/tests/book/test_inference_nlp.cc
浏览文件 @
602e28bf
...
...
@@ -12,6 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */
#include <sys/time.h>
#include <time.h>
#include "gflags/gflags.h"
#include "gtest/gtest.h"
#include "paddle/fluid/inference/tests/test_helper.h"
...
...
@@ -29,32 +31,28 @@ TEST(inference, understand_sentiment) {
// 0. Call `paddle::framework::InitDevices()` initialize all the devices
// In unittests, this is done in paddle/testing/paddle_gtest_main.cc
paddle
::
framework
::
LoDTensor
words
;
paddle
::
framework
::
LoD
lod
{{
0
,
83
}};
int64_t
word_dict_len
=
198392
;
SetupLoDTensor
(
&
words
,
lod
,
static_cast
<
int64_t
>
(
0
),
static_cast
<
int64_t
>
(
word_dict_len
-
1
));
/*
std::vector<int64_t> srcdata{
784, 784, 1550, 6463, 56, 75693, 6189, 784, 784, 1550,
198391, 6463, 42468, 4376, 10251, 10760, 6189, 297, 396, 6463,
6463, 1550, 198391, 6463, 22564, 1612, 291, 68, 164, 784,
784, 1550, 198391, 6463, 13659, 3362, 42468, 6189, 2209,
198391,
6463, 2209, 2209, 198391, 6463, 2209, 1062, 3029, 1831, 3029,
1065, 2281, 100, 11216, 1110, 56, 10869, 9811, 100,
198391,
6463, 100, 9280, 100, 288, 40031, 1680, 1335, 100, 1550,
9280, 7265, 244, 1550, 198391, 6463, 1550, 198391, 6463,
42468,
4376, 10251, 10760};
paddle::framework::LoD lod{{0, srcdata.size()}};
words.set_lod(lod);
int64_t* pdata =
words.mutable_data<int64_t>({static_cast<int64_t>(srcdata.size()), 1},
paddle::platform::CPUPlace());
memcpy(pdata, srcdata.data(), words.numel() * sizeof(int64_t));
*/
paddle::framework::LoD lod{{0, 83}};
int64_t word_dict_len = 198392;
SetupLoDTensor(&words, lod, static_cast<int64_t>(0),
static_cast<int64_t>(word_dict_len - 1));
*/
std
::
vector
<
int64_t
>
srcdata
{
784
,
784
,
1550
,
6463
,
56
,
75693
,
6189
,
784
,
784
,
1550
,
198391
,
6463
,
42468
,
4376
,
10251
,
10760
,
6189
,
297
,
396
,
6463
,
6463
,
1550
,
198391
,
6463
,
22564
,
1612
,
291
,
68
,
164
,
784
,
784
,
1550
,
198391
,
6463
,
13659
,
3362
,
42468
,
6189
,
2209
,
198391
,
6463
,
2209
,
2209
,
198391
,
6463
,
2209
,
1062
,
3029
,
1831
,
3029
,
1065
,
2281
,
100
,
11216
,
1110
,
56
,
10869
,
9811
,
100
,
198391
,
6463
,
100
,
9280
,
100
,
288
,
40031
,
1680
,
1335
,
100
,
1550
,
9280
,
7265
,
244
,
1550
,
198391
,
6463
,
1550
,
198391
,
6463
,
42468
,
4376
,
10251
,
10760
};
paddle
::
framework
::
LoD
lod
{{
0
,
srcdata
.
size
()}};
words
.
set_lod
(
lod
);
int64_t
*
pdata
=
words
.
mutable_data
<
int64_t
>
(
{
static_cast
<
int64_t
>
(
srcdata
.
size
()),
1
},
paddle
::
platform
::
CPUPlace
());
memcpy
(
pdata
,
srcdata
.
data
(),
words
.
numel
()
*
sizeof
(
int64_t
));
LOG
(
INFO
)
<<
"number of input size:"
<<
words
.
numel
();
std
::
vector
<
paddle
::
framework
::
LoDTensor
*>
cpu_feeds
;
cpu_feeds
.
push_back
(
&
words
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录