experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

address review comments.

Browse Source
This commit is contained in:
Advait Jain 2021-02-09 14:44:06 -08:00
parent b885fb44dd
commit 9f5f278e83
4 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/micro/BUILD
View File

@ -248,6 +248,7 @@ tflite_micro_cc_test(
"micro_interpreter_test.cc", "micro_interpreter_test.cc",
], ],
deps = [ deps = [
":micro_compatibility",
":micro_framework", ":micro_framework",
":micro_utils", ":micro_utils",
":op_resolvers", ":op_resolvers",

6
tensorflow/lite/micro/benchmarks/keyword_benchmark.cc
View File

@ -84,14 +84,14 @@ int main(int argc, char** argv) {
CreateBenchmarkRunner(&profiler); CreateBenchmarkRunner(&profiler);
profiler.EndEvent(event_handle); profiler.EndEvent(event_handle);
profiler.Log(); profiler.Log();
MicroPrintf(""); MicroPrintf(""); // null MicroPrintf serves as a newline.
tflite::KeywordRunNIerations(1, "KeywordRunNIerations(1)", *benchmark_runner, tflite::KeywordRunNIerations(1, "KeywordRunNIerations(1)", *benchmark_runner,
profiler); profiler);
profiler.Log(); profiler.Log();
MicroPrintf(""); MicroPrintf(""); // null MicroPrintf serves as a newline.
tflite::KeywordRunNIerations(10, "KeywordRunNIerations(10)", tflite::KeywordRunNIerations(10, "KeywordRunNIerations(10)",
*benchmark_runner, profiler); *benchmark_runner, profiler);
MicroPrintf(""); MicroPrintf(""); // null MicroPrintf serves as a newline.
} }

3
tensorflow/lite/micro/micro_interpreter_test.cc
View File

@ -19,6 +19,7 @@ limitations under the License.
#include "tensorflow/lite/core/api/flatbuffer_conversions.h" #include "tensorflow/lite/core/api/flatbuffer_conversions.h"
#include "tensorflow/lite/micro/all_ops_resolver.h" #include "tensorflow/lite/micro/all_ops_resolver.h"
#include "tensorflow/lite/micro/compatibility.h"
#include "tensorflow/lite/micro/micro_error_reporter.h" #include "tensorflow/lite/micro/micro_error_reporter.h"
#include "tensorflow/lite/micro/micro_profiler.h" #include "tensorflow/lite/micro/micro_profiler.h"
#include "tensorflow/lite/micro/micro_utils.h" #include "tensorflow/lite/micro/micro_utils.h"
@ -46,6 +47,8 @@ class MockProfiler : public MicroProfiler {
private: private:
int event_starts_; int event_starts_;
int event_ends_; int event_ends_;
TF_LITE_REMOVE_VIRTUAL_DELETE
}; };
} // namespace } // namespace

6
tensorflow/lite/micro/micro_profiler.h
View File

@ -18,6 +18,8 @@ limitations under the License.
#include <cstdint> #include <cstdint>
#include "tensorflow/lite/micro/compatibility.h"
namespace tflite { namespace tflite {
// MicroProfiler creates a common way to gain fine-grained insight into runtime // MicroProfiler creates a common way to gain fine-grained insight into runtime
@ -27,7 +29,7 @@ namespace tflite {
class MicroProfiler { class MicroProfiler {
public: public:
MicroProfiler() = default; MicroProfiler() = default;
~MicroProfiler() = default; virtual ~MicroProfiler() = default;
// Marks the start of a new event and returns an event handle that can be used // Marks the start of a new event and returns an event handle that can be used
// to mark the end of the event via EndEvent. The lifetime of the tag // to mark the end of the event via EndEvent. The lifetime of the tag
@ -64,6 +66,8 @@ class MicroProfiler {
int32_t start_ticks_[kMaxEvents]; int32_t start_ticks_[kMaxEvents];
int32_t end_ticks_[kMaxEvents]; int32_t end_ticks_[kMaxEvents];
int num_events_ = 0; int num_events_ = 0;
TF_LITE_REMOVE_VIRTUAL_DELETE;
}; };
#if defined(NDEBUG) #if defined(NDEBUG)