Searched defs:algorithm_search (Results 1 – 2 of 2) sorted by relevance
167 struct algorithm_search {}; struct260 struct algorithm_search<cudnnConvolutionFwdAlgoPerf_t> { struct261 using perf_t = cudnnConvolutionFwdAlgoPerf_t;262 using algo_t = cudnnConvolutionFwdAlgo_t;264 static constexpr auto DEFAULT_ALGO =266 static BenchmarkCache<perf_t>& cache() { in cache()270 static std::vector<perf_t> findAlgorithms( in findAlgorithms()330 static void getWorkspaceSize( in getWorkspaceSize()348 struct algorithm_search<cudnnConvolutionBwdDataAlgoPerf_t> { struct349 using perf_t = cudnnConvolutionBwdDataAlgoPerf_t;[all …]
310 struct algorithm_search { struct359 struct algorithm_search<miopenConvFwdAlgorithm_t> { struct360 using perf_t = miopenConvAlgoPerf_t;361 using algo_t = miopenConvFwdAlgorithm_t;363 static constexpr auto DEFAULT_ALGO = miopenConvolutionFwdAlgoGEMM;364 static BenchmarkCache<algo_t>& cache() { return fwd_algos; } in cache()365 static BenchmarkCache<size_t>& wsscache() { return fwd_wssizes; } in wsscache()367 static perf_t findAlgorithm(const ConvolutionArgs& args) { in findAlgorithm()387 static miopenConvSolution_t getSolution(const ConvolutionArgs& args, bool force_default) { in getSolution()432 struct algorithm_search<miopenConvBwdDataAlgorithm_t> { struct[all …]