Searched refs:algo_t (Results 1 – 2 of 2) sorted by relevance
209 template <typename algo_t>212 const algo_t* algo, in getMaxWorkspaceSize()262 using algo_t = cudnnConvolutionFwdAlgo_t; typedef273 static const algo_t algos[] = { in findAlgorithms()332 algo_t algo, in getWorkspaceSize()350 using algo_t = cudnnConvolutionBwdDataAlgo_t; typedef360 static const algo_t algos[] = { in findAlgorithms()434 using algo_t = cudnnConvolutionBwdFilterAlgo_t; typedef445 static const algo_t algos[] = { in findAlgorithms()504 algo_t algo, in getWorkspaceSize()
309 template<typename algo_t>361 using algo_t = miopenConvFwdAlgorithm_t; typedef364 static BenchmarkCache<algo_t>& cache() { return fwd_algos; } in cache()434 using algo_t = miopenConvBwdDataAlgorithm_t; typedef437 static BenchmarkCache<algo_t>& cache() { return bwd_data_algos; } in cache()507 using algo_t = miopenConvBwdWeightsAlgorithm_t; typedef510 static BenchmarkCache<algo_t>& cache() { return bwd_filter_algos; } in cache()577 template<typename algo_t>578 void findAlgorithm(const ConvolutionArgs& args, bool benchmark, algo_t* algo) { in findAlgorithm()579 using search = algorithm_search<algo_t>; in findAlgorithm()[all …]