Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
weixin_51669992
tensorflow
提交
207bd43d
T
tensorflow
项目概览
weixin_51669992
/
tensorflow
与 Fork 源项目一致
从无法访问的项目Fork
通知
16
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
tensorflow
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
207bd43d
编写于
8月 12, 2019
作者:
C
Chao Mei
提交者:
TensorFlower Gardener
8月 12, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add a HasProfiles API to simplify the profile-empty check in the benchmark tool.
PiperOrigin-RevId: 262869158
上级
f9233a58
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
4 addition
and
6 deletion
+4
-6
tensorflow/lite/profiling/profile_summarizer.h
tensorflow/lite/profiling/profile_summarizer.h
+2
-0
tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
+2
-6
未找到文件。
tensorflow/lite/profiling/profile_summarizer.h
浏览文件 @
207bd43d
...
...
@@ -45,6 +45,8 @@ class ProfileSummarizer {
return
stats_calculator_
->
GetShortSummary
();
}
bool
HasProfiles
()
const
{
return
stats_calculator_
->
num_runs
()
>=
1
;
}
private:
std
::
unique_ptr
<
tensorflow
::
StatsCalculator
>
stats_calculator_
;
};
...
...
tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
浏览文件 @
207bd43d
...
...
@@ -56,9 +56,7 @@ constexpr int kOpProfilingEnabledDefault = false;
class
ProfilingListener
:
public
BenchmarkListener
{
public:
explicit
ProfilingListener
(
Interpreter
*
interpreter
,
uint32_t
max_num_entries
)
:
interpreter_
(
interpreter
),
profiler_
(
max_num_entries
),
has_profiles_
(
false
)
{
:
interpreter_
(
interpreter
),
profiler_
(
max_num_entries
)
{
TFLITE_BENCHMARK_CHECK
(
interpreter
);
interpreter_
->
SetProfiler
(
&
profiler_
);
}
...
...
@@ -73,7 +71,6 @@ class ProfilingListener : public BenchmarkListener {
Interpreter
*
interpreter_
;
profiling
::
BufferedProfiler
profiler_
;
profiling
::
ProfileSummarizer
summarizer_
;
bool
has_profiles_
;
};
// Dumps gemmlowp profiling events if gemmlowp profiling is enabled.
...
...
@@ -92,7 +89,7 @@ void ProfilingListener::OnSingleRunStart(RunType run_type) {
}
void
ProfilingListener
::
OnBenchmarkEnd
(
const
BenchmarkResults
&
results
)
{
if
(
has_profiles_
)
{
if
(
summarizer_
.
HasProfiles
()
)
{
TFLITE_LOG
(
INFO
)
<<
summarizer_
.
GetOutputString
();
}
}
...
...
@@ -100,7 +97,6 @@ void ProfilingListener::OnBenchmarkEnd(const BenchmarkResults& results) {
void
ProfilingListener
::
OnSingleRunEnd
()
{
profiler_
.
StopProfiling
();
auto
profile_events
=
profiler_
.
GetProfileEvents
();
has_profiles_
=
!
profile_events
.
empty
();
summarizer_
.
ProcessProfiles
(
profile_events
,
*
interpreter_
);
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录