请问 tensorflow 中如何确定是否使用 GEMM 计算卷积?
#if !defined (USE_GEMM_FOR_CONV)
…
#endif // USE_GEMM_FOR_CONV
如何确定是否 define 了 USE_GEMMFOR_CONV
提问人:light,发表在 2018-10-11 22:24:04
请问 tensorflow 中如何确定是否使用 GEMM 计算卷积?
#if !defined (USE_GEMM_FOR_CONV)
…
#endif // USE_GEMM_FOR_CONV
如何确定是否 define 了 USE_GEMMFOR_CONV
提问人:light,发表在 2018-10-11 22:24:04
应该是编译的时候的一个参数
舟 3332 发表于 2018-10-15
谢谢您
可以使用 bazel 编译指定这个参数吗?一旦编译完成这个参数就确定了,会随使用过程改变吗?
light,2018-10-16 08:40
这种 #define 都是编译时候的参数。运行时无法改变呢。
bazel build --copt=“-DUSE_GEMM_FOR_CONV” 就应该可以
舟 3332,2018-10-17 00:36