发新帖

tensorflow卷积实现问题

[复制链接]
195 3

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
请问tensorflow中如何确定是否使用GEMM 计算卷积?

#if !defined(USE_GEMM_FOR_CONV)
...
#endif  // USE_GEMM_FOR_CONV
如何确定是否define了USE_GEMMFOR_CONV


舟3332已获得悬赏 10 金币+15 金币

最佳答案

应该是编译的时候的一个参数
本楼点评(0) 收起

精彩评论3

舟3332  TF芽芽  发表于 2018-10-15 18:28:15 来自手机  | 显示全部楼层
应该是编译的时候的一个参数
本楼点评(2) 收起
  • light谢谢您
    可以使用bazel 编译指定这个参数吗?一旦编译完成这个参数就确定了,会随使用过程改变吗?
    2018-10-16 08:40 回复
  • 舟3332回复 light :  这种 #define 都是编译时候的参数。运行时无法改变呢。


    bazel build --copt="-DUSE_GEMM_FOR_CONV" 就应该可以
    2018-10-17 00:36 回复
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

6

积分
快速回复 返回顶部 返回列表