Home
last modified time | relevance | path

Searched defs:dimf (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DConvolutionMM2d.cu46 constexpr int dimf = 1; in slow_conv2d_shape_check() local
137 constexpr int dimf = 1; in slow_conv2d_forward() local
H A DNaiveConvolutionTranspose2d.cu95 int dimf = 0; in slow_conv_transpose2d_shape_check() local
H A DNaiveConvolutionTranspose3d.cu113 int dimf = 0; in slow_conv_transpose3d_shape_check() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveConvolutionTranspose2d.cpp92 int dimf = 0; in slow_conv_transpose2d_shape_check() local
H A DNaiveConvolutionTranspose3d.cpp114 int dimf = 0; in slow_conv_transpose3d_shape_check() local