Searched refs:interp_width (Results 1 – 2 of 2) sorted by relevance
502 const int interp_width = (int)ceilf(support_w) * 2 + 1; in upsample_gen2d_aa_out_frame() local506 scalar_t* wx = reinterpret_cast<scalar_t*>(smem) + interp_width * threadIdx.x; in upsample_gen2d_aa_out_frame()507 …scalar_t* wy = reinterpret_cast<scalar_t*>(smem) + interp_width * blockDim.x + interp_height * thr… in upsample_gen2d_aa_out_frame()508 const int offset = interp_width * blockDim.x + interp_height * blockDim.y; in upsample_gen2d_aa_out_frame()526 interp_width, in upsample_gen2d_aa_out_frame()608 const int interp_width = (int)ceilf(support_w) * 2 + 1; in upsample_gen2d_aa_backward_out_frame() local612 scalar_t* wx = reinterpret_cast<scalar_t*>(smem) + interp_width * threadIdx.x; in upsample_gen2d_aa_backward_out_frame()613 …scalar_t* wy = reinterpret_cast<scalar_t*>(smem) + interp_width * blockDim.x + interp_height * thr… in upsample_gen2d_aa_backward_out_frame()629 interp_width, in upsample_gen2d_aa_backward_out_frame()727 const int interp_width = 1 + 2 * (int)ceilf( in upsample_gen2d_aa_out_cuda_template() local[all …]
1970 const int interp_width = (int)ceilf(support_w) * 2 + 1; in cpu_upsample_genNd_backward_aa() local1972 std::vector<scalar_t> wx(interp_width, 0.0); in cpu_upsample_genNd_backward_aa()2000 interp_width, in cpu_upsample_genNd_backward_aa()