Searched defs:grad_warp (Results 1 – 2 of 2) sorted by relevance
147 T* __restrict__ grad_warp, const int batch_size, const int data_height, in ResamplerGrad2DKernel()
662 auto grad_warp = in Compile() local