Определяет ли тензорный поток USE_GEMM_FOR_CONV - PullRequest
0 голосов
/ 11 октября 2018

Орудие сверточной операции в 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?

...