Home
last modified time | relevance | path

Searched refs:BenchmarkCache (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/cudnn/
DConv_v7.cpp113 struct BenchmarkCache { struct
138 BenchmarkCache<cudnnConvolutionFwdAlgoPerf_t> fwd_algos;
139 BenchmarkCache<cudnnConvolutionBwdDataAlgoPerf_t> bwd_data_algos;
140 BenchmarkCache<cudnnConvolutionBwdFilterAlgoPerf_t> bwd_filter_algos;
266 static BenchmarkCache<perf_t>& cache() { in cache()
353 static BenchmarkCache<perf_t>& cache() { in cache()
438 static BenchmarkCache<perf_t>& cache() { in cache()
DConv_v8.cpp286 struct BenchmarkCache { struct
351 thread_local BenchmarkCache<cudnn_frontend::ExecutionPlan, CacheKeyWrapper>
353 thread_local BenchmarkCache<cudnn_frontend::ExecutionPlan, CacheKeyFusedWrapper>
/external/pytorch/aten/src/ATen/native/miopen/
DConv_miopen.cpp264 struct BenchmarkCache { struct
284 BenchmarkCache<miopenConvFwdAlgorithm_t> fwd_algos;
285 BenchmarkCache<miopenConvBwdDataAlgorithm_t> bwd_data_algos;
286 BenchmarkCache<miopenConvBwdWeightsAlgorithm_t> bwd_filter_algos;
288 BenchmarkCache<size_t> fwd_wssizes;
289 BenchmarkCache<size_t> bwd_data_wssizes;
290 BenchmarkCache<size_t> bwd_filter_wssizes;
364 static BenchmarkCache<algo_t>& cache() { return fwd_algos; } in cache()
365 static BenchmarkCache<size_t>& wsscache() { return fwd_wssizes; } in wsscache()
437 static BenchmarkCache<algo_t>& cache() { return bwd_data_algos; } in cache()
[all …]