Как я могу выяснить, использует ли тензор поток GEMM для сверточной операции? - PullRequest
0 голосов
/ 03 октября 2018

Я хочу выяснить процесс выполнения сверточной операции tenorflow.Я обнаружил, что операция была реализована в файле 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

Как я могу понятьвне зависимости от того, использует ли tenorflow GEMM для сверточной операцииСпасибо за любую помощь.

...