如何确定张量流是否使用GEMM进行卷积运算?

时间:2018-10-03 14:09:58

标签: c++ tensorflow convolution

我想弄清楚张量流的卷积运算的执行过程,发现该运算是在conv_ops.cc文件中实现的,但是我有些困惑。

#if !defined(USE_GEMM_FOR_CONV)
TF_CALL_half(REGISTER_CPU);
TF_CALL_float(REGISTER_CPU);
TF_CALL_double(REGISTER_CPU);
#endif  // USE_GEMM_FOR_CONV

如何确定张量流是否使用GEMM进行卷积运算? 感谢您的帮助。

0 个答案:

没有答案