标签: 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进行卷积运算? 感谢您的帮助。