Searched refs:algo_types (Results 1 – 2 of 2) sorted by relevance
5283 std::vector<dnn::AlgorithmDesc::Index> algo_types; in GetConvolveAlgorithms() local5286 algo_types = {CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM}; in GetConvolveAlgorithms()5288 algo_types = {CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM, in GetConvolveAlgorithms()5295 algo_types.push_back(CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING); in GetConvolveAlgorithms()5298 algo_types.push_back(CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD_NONFUSED); in GetConvolveAlgorithms()5303 for (auto i : algo_types) { in GetConvolveAlgorithms()5317 std::vector<dnn::AlgorithmDesc::Index> algo_types = { in GetRnnAlgorithms() local5326 for (auto i : algo_types) { in GetRnnAlgorithms()5342 std::vector<dnn::AlgorithmDesc::Index> algo_types = { in GetConvolveBackwardDataAlgorithms() local5351 algo_types.push_back(CUDNN_CONVOLUTION_BWD_DATA_ALGO_WINOGRAD_NONFUSED); in GetConvolveBackwardDataAlgorithms()[all …]
146 static const struct typeent algo_types[] = { variable158 const struct typeent *t = &algo_types[i]; in xfrm_algotype_getbyname()176 const struct typeent *t = &algo_types[i]; in strxf_algotype()