Searched defs:gI_stride (Results 1 – 2 of 2) sorted by relevance
58 const auto gI_stride = grad_input.strides()[wrap_dim] * dim_size; in glu_backward_cuda_out() local
111 int64_t gI_stride, in launch_glu_backward_kernel()