Searched refs:fft_plan_cache (Results 1 – 3 of 3) sorted by relevance
98 int device_ordinal, FftPlanCache* fft_plan_cache, in RunFft() argument108 FftPlan* fft_plan_ptr = fft_plan_cache->GetOrCreate(device_ordinal); in RunFft()
94 int device_ordinal, FftPlanCache* fft_plan_cache,
1250 FftPlanCache fft_plan_cache; in operator ()() local1279 &fft_plan_cache, stream, run_options->allocator()); in operator ()()