Searched defs:k_row (Results 1 – 1 of 1) sorted by relevance
73 for (int k_row = 0; k_row < kH; ++k_row) { in conv_depthwise3d_cuda_kernel() local150 for (int k_row = 0; k_row < kH; ++k_row) { in conv_depthwise3d_cuda_backward_input_kernel() local198 const int k_row = (blockIdx.x / kW) % kH; in conv_depthwise3d_cuda_backward_weight_kernel() local