Searched refs:chooseAlgorithm (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | Conv_miopen.cpp | 609 Workspace chooseAlgorithm( in chooseAlgorithm() function
729 Workspace workspace = chooseAlgorithm(args, benchmark, &fwdAlg); in raw_miopen_convolution_forward_out()
837 Workspace workspace = chooseAlgorithm(args, benchmark, &fwdAlg); in raw_miopen_depthwise_convolution_forward_out()
992 Workspace workspace = chooseAlgorithm(args, benchmark, &bwdFilterAlg); in raw_miopen_convolution_backward_weight_out()
1036 Workspace workspace = chooseAlgorithm(args, benchmark, &bwdFilterAlg); in raw_miopen_depthwise_convolution_backward_weight_out()
1241 Workspace workspace = chooseAlgorithm(args, benchmark, &bwdDataAlg); in raw_miopen_convolution_backward_input_out()
1350 Workspace workspace = chooseAlgorithm(args, benchmark, &bwdDataAlg); in raw_miopen_depthwise_convolution_backward_input_out()
|