张量流是否定义USE_GEMM_FOR_CONV

时间:2018-10-11 15:11:55

标签: c++ tensorflow conv-neural-network convolution

conv_ops.cc中卷积运算的实现涉及USE_GEMM_FOR_CONV

#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

如何确定程序是否定义了USE_GEMM_FOR_CONV

0 个答案:

没有答案