Searched defs:k_col (Results 1 – 1 of 1) sorted by relevance
75 for (int k_col = 0; k_col < kW; ++k_col) { in conv_depthwise3d_cuda_kernel() local153 for (int k_col = 0; k_col < kW; ++k_col) { in conv_depthwise3d_cuda_backward_input_kernel() local197 const int k_col = blockIdx.x % kW; in conv_depthwise3d_cuda_backward_weight_kernel() local