Searched defs:iW (Results 1 – 5 of 5) sorted by relevance
49 int64_t iW, // input sizes in spatial_dilated_max_pooling()108 int64_t iW, // input sizes in spatial_dilated_max_pooling3d()227 int64_t iW = qx.size(dimw); in q_maxpool_2d() local393 int64_t iW = qx.size(dimw); in q_maxpool_3d() local
47 const int iW = input.size(4); in conv_depthwise3d_cuda_kernel() local118 const int iW = grad_input.size(4); in conv_depthwise3d_cuda_backward_input_kernel() local208 const int iW = input.size(4); in conv_depthwise3d_cuda_backward_weight_kernel() local
242 int iW = (width + gridDim.y-1) / gridDim.y; in C10_LAUNCH_BOUNDS_1() local
396 index_t iW = (isizeW + gridDim.y-1) / gridDim.y; in C10_LAUNCH_BOUNDS_1() local
1456 int64_t iW, // input sizes in _qmaxpool_2d_nhwc_kernel()1556 int64_t iW, // input sizes in qmaxpool_2d_nhwc_kernel()1584 int64_t iW, // input sizes in qmaxpool_3d_nthwc_kernel()