Searched refs:DWCONV3D_FORWARD_DISPATCH_SPECIALIZATION (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DepthwiseConv3d.cu | 357 #define DWCONV3D_FORWARD_DISPATCH_SPECIALIZATION(kt, kh, kw, dilt, dilh, dilw) \ macro
452 DWCONV3D_FORWARD_DISPATCH_SPECIALIZATION(3, 3, 3, 1, 1, 1) in conv_depthwise3d_cuda()
453 DWCONV3D_FORWARD_DISPATCH_SPECIALIZATION(-1, -1, -1, 1, 1, 1) in conv_depthwise3d_cuda()
461 #undef DWCONV3D_FORWARD_DISPATCH_SPECIALIZATION
|