diff options
author | 2018-07-16 12:51:48 -0700 | |
---|---|---|
committer | 2018-07-16 12:52:18 -0700 | |
commit | fbffff0b83d04675b9ce64d9e1fb573a79521cd8 (patch) | |
tree | 417b852521c0a672ffa5891ea4264765fce43ff9 | |
parent | 814f9ccd9b34a828f93d33eee6265e0cac07095a (diff) | |
parent | 6c4ffe1bee9e7f8e1b181680251898bc842260b1 (diff) |
Merge pull request #20330 from freedomtan:fix_benchmark_model_tflite
PiperOrigin-RevId: 204789631
-rw-r--r-- | tensorflow/contrib/lite/profiling/profile_summarizer.cc | 2 | ||||
-rw-r--r-- | tensorflow/contrib/lite/tools/benchmark/benchmark_params.h | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/profiling/profile_summarizer.cc b/tensorflow/contrib/lite/profiling/profile_summarizer.cc index c37a096588..36e87b666a 100644 --- a/tensorflow/contrib/lite/profiling/profile_summarizer.cc +++ b/tensorflow/contrib/lite/profiling/profile_summarizer.cc @@ -83,7 +83,7 @@ OperatorDetails GetOperatorDetails(const tflite::Interpreter& interpreter, OperatorDetails details; details.name = op_name; if (profiling_string) { - details.name += ":" + string(profiling_string); + details.name += ":" + std::string(profiling_string); } details.inputs = GetTensorNames(interpreter, inputs); details.outputs = GetTensorNames(interpreter, outputs); diff --git a/tensorflow/contrib/lite/tools/benchmark/benchmark_params.h b/tensorflow/contrib/lite/tools/benchmark/benchmark_params.h index 33448dd162..c98f47bb0d 100644 --- a/tensorflow/contrib/lite/tools/benchmark/benchmark_params.h +++ b/tensorflow/contrib/lite/tools/benchmark/benchmark_params.h @@ -31,6 +31,8 @@ class TypedBenchmarkParam; class BenchmarkParam { protected: enum class ParamType { TYPE_INT32, TYPE_FLOAT, TYPE_BOOL, TYPE_STRING }; + template <typename T> + static ParamType GetValueType(); public: template <typename T> @@ -49,8 +51,6 @@ class BenchmarkParam { private: static void AssertHasSameType(ParamType a, ParamType b); - template <typename T> - static ParamType GetValueType(); const ParamType type_; }; |