Home
last modified time | relevance | path

Searched refs:select_conv_backend (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvUtils.h124 TORCH_API ConvBackend select_conv_backend(
H A DConvolution.cpp1311 ConvBackend select_conv_backend( in select_conv_backend() function
1354 static ConvBackend select_conv_backend( in select_conv_backend() function
2045 …ConvBackend backend = select_conv_backend(input, weight, bias_sizes_opt, /*need_backward=*/ true, … in convolution_backward()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1920 return at::native::select_conv_backend( in initModule()
1959 return at::native::select_conv_backend( in initModule()