Home | History | Annotate | Download | only in Tensor

Lines Matching refs:numPlanes

575     const float* __restrict kernel, const int numPlanes, const int numX,
587 for (int p = first_plane + threadIdx.y; p < numPlanes; p += plane_stride) {
623 const float* __restrict kernel, const int numPlanes, const int numX,
641 for (int p = first_plane + threadIdx.z; p < numPlanes; p += plane_stride) {
690 const float* __restrict kernel, const size_t numPlanes, const size_t numX,
709 for (int p = 0; p < numPlanes; ++p) {