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

[TFLite] Introduce ruy_profiler to makefile

Browse Source 
Introduce `BUILD_WITH_RUY_PROFILER` option to support ruy_profiler
This commit is contained in:
periannath 2020-04-20 21:49:25 +09:00
parent ccd260cd82
commit 3e6030a241
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/lite/tools/make/Makefile
View File

@ -184,6 +184,14 @@ ifeq ($(BUILD_WITH_RUY),true)
CXXFLAGS += -DTFLITE_WITH_RUY CXXFLAGS += -DTFLITE_WITH_RUY
endif endif
BUILD_WITH_RUY_PROFILER ?= false
ifeq ($(BUILD_WITH_RUY_PROFILER),true)
CORE_CC_ALL_SRCS += tensorflow/lite/tools/make/downloads/ruy/ruy/profiler/instrumentation.cc
CORE_CC_ALL_SRCS += tensorflow/lite/tools/make/downloads/ruy/ruy/profiler/profiler.cc
CORE_CC_ALL_SRCS += tensorflow/lite/tools/make/downloads/ruy/ruy/profiler/treeview.cc
CXXFLAGS += -DRUY_PROFILER
endif
# Not to include XNNPACK. # Not to include XNNPACK.
CXXFLAGS += -DTFLITE_WITHOUT_XNNPACK CXXFLAGS += -DTFLITE_WITHOUT_XNNPACK