Как получить ручку кубласа в тензорном потоке? - PullRequest
0 голосов
/ 29 сентября 2019

Я пишу пользовательскую операцию, используя функции cublas cublasCgetrfBatched и cublasCgetriBatched, функции используют дескриптор cublas в качестве входного параметра, однако cublasCreate(&handle); стоит почти 100 мс.Я думаю, что TF уже интегрировал модуль CUBLAS, cublasCreate(&handle) должно быть вызвано в процессе инициализации, тогда как получить дескриптор?Пример может быть лучшим!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...