本文档介绍了如何在使用Bazel构建TensorFlow Lite时定义预处理器宏`TF_LITE_STATIC_MEMORY`。通过创建一个名为`tflite_static_memory_enabled`的`cc_library`并设置`defines`为`["TF_LITE_STATIC_MEMORY"]`，然后在需要使用该宏的模块中添加对此库的依赖，可以实现`TF_LITE_STATIC_MEMORY`的定义。
tensorflow是用bazel编译的，如果代码中出现如下 #ifdef TF_LITE_STATIC_MEMORY，怎样去定义TF_LITE_STATIC_MEMORY？
如果是makefile， CCFLAGS := -DNDEBUG -g -DTF_LITE_STATIC_MEMORY
google的结果是有人提出这样的问题，但没有人给出显示的答案，但看到defines可能相关 ，在代码中搜索找到
lite/kernels/conv.cc
cc_library(
name = "tflite_with_ruy_enabled",
defines = ["TFLITE_WITH_RUY"],
visibility = ["//visibility:private"],
)
源码中对应的使用是：
TfLiteRegistration* Register_CONV_2D() {
#if defined TFLITE_USE_APPLE_ACCELERATE_FOR_CONV
return Register_CONVOLUTION_CBLAS_OPT();
#elif defined TFLITE_WITH_RUY
// tflite_with_ruy optimizes the generic kern
最低0.47元/天 解锁文章
扫一扫
1491
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
