Home
last modified time | relevance | path

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

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