Searched refs:algo_types (Results 1 – 2 of 2) sorted by relevance
4417 std::vector<dnn::AlgorithmDesc::Index> algo_types; in GetConvolveAlgorithms() local4420 algo_types = {CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM}; in GetConvolveAlgorithms()4422 algo_types = {CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM, in GetConvolveAlgorithms()4429 algo_types.push_back(CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING); in GetConvolveAlgorithms()4432 algo_types.push_back(CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD_NONFUSED); in GetConvolveAlgorithms()4437 for (auto i : algo_types) { in GetConvolveAlgorithms()4456 std::vector<dnn::AlgorithmDesc::Index> algo_types = { in GetRnnAlgorithms() local4465 for (auto i : algo_types) { in GetRnnAlgorithms()4486 std::vector<dnn::AlgorithmDesc::Index> algo_types = { in GetConvolveBackwardDataAlgorithms() local4495 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()