Tensorflow 卷积实现问题

请问 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