Searched defs:start_feature (Results 1 – 4 of 4) sorted by relevance
690 int start_feature = threadIdx.y * 4; in EigenFloatContractionKernelInternal16x16() local1062 int start_feature = (threadIdx.y / 4) * 8; in EigenFloatContractionKernelInternal() local
160 const int start_feature = threadIdx.x + blockIdx.y * blockDim.x * SZ; in embedding_backward_kernel() local
77 int64_t start_feature = threadIdx.x + blockIdx.y * blockDim.x * SZ; in indexing_backward_kernel() local240 int64_t start_feature = threadIdx.x + blockIdx.y * blockDim.x * SZ; in indexing_backward_kernel_quantized() local
1029 def start_feature(self, location, name): member in Builder