HomeSort by relevance Sort by last modified time
    Searched full:cuda_fft_plan (Results 1 - 1 of 1) sorted by null

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_fft.cc 454 CUDAFftPlan *cuda_fft_plan = dynamic_cast<CUDAFftPlan *>(plan); local
456 cuda_fft_plan->UpdateScratchAllocator(stream, scratch_allocator);
467 CUDAFftPlan *cuda_fft_plan = dynamic_cast<CUDAFftPlan *>(plan); local
468 if (cuda_fft_plan == nullptr) {
473 if (!SetStream(parent_, cuda_fft_plan->GetPlan(), stream)) {
477 auto ret = cufftExec(parent_, cuda_fft_plan->GetPlan(),
494 CUDAFftPlan *cuda_fft_plan = dynamic_cast<CUDAFftPlan *>(plan); local
495 if (cuda_fft_plan == nullptr) {
500 if (!SetStream(parent_, cuda_fft_plan->GetPlan(), stream)) {
504 auto ret = cufftExec(parent_, cuda_fft_plan->GetPlan()
    [all...]

Completed in 89 milliseconds