Searched refs:algo_types (Results 1 – 2 of 2) sorted by relevance
3305 std::vector<dnn::AlgorithmDesc::Index> algo_types; in GetConvolveAlgorithms() local3308 algo_types = {CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM}; in GetConvolveAlgorithms()3310 algo_types = { in GetConvolveAlgorithms()3321 algo_types.push_back(CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING); in GetConvolveAlgorithms()3324 algo_types.push_back(CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD_NONFUSED); in GetConvolveAlgorithms()3329 for (auto i : algo_types) { in GetConvolveAlgorithms()3348 std::vector<dnn::AlgorithmDesc::Index> algo_types = { in GetRnnAlgorithms() local3357 for (auto i : algo_types) { in GetRnnAlgorithms()3377 std::vector<dnn::AlgorithmDesc::Index> algo_types = { in GetConvolveBackwardDataAlgorithms() local3386 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()