From 2307db76a2a07c7af6581e0ef4c6a5a0b83921f4 Mon Sep 17 00:00:00 2001 From: Shashi Shekhar Date: Wed, 23 May 2018 17:14:39 -0700 Subject: Refactor StatSummarizer extract common functionality without proto dependencies. PiperOrigin-RevId: 197816405 --- tensorflow/contrib/android/jni/run_stats_jni.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'tensorflow/contrib/android') diff --git a/tensorflow/contrib/android/jni/run_stats_jni.cc b/tensorflow/contrib/android/jni/run_stats_jni.cc index 707853b59b..30de7b59af 100644 --- a/tensorflow/contrib/android/jni/run_stats_jni.cc +++ b/tensorflow/contrib/android/jni/run_stats_jni.cc @@ -16,6 +16,7 @@ limitations under the License. #include "tensorflow/contrib/android/jni/run_stats_jni.h" #include + #include #include "tensorflow/core/protobuf/config.pb.h" @@ -73,7 +74,8 @@ JNIEXPORT jstring RUN_STATS_METHOD(summary)(JNIEnv* env, jclass clazz, StatSummarizer* s = requireHandle(env, handle); if (s == nullptr) return nullptr; std::stringstream ret; - ret << s->GetStatsByMetric("Top 10 CPU", StatSummarizer::BY_TIME, 10) + ret << s->GetStatsByMetric("Top 10 CPU", tensorflow::StatsCalculator::BY_TIME, + 10) << s->GetStatsByNodeType() << s->ShortSummary(); return env->NewStringUTF(ret.str().c_str()); } -- cgit v1.2.3