Searched defs:featureDim (Results 1 – 3 of 3) sorted by relevance
222 int featureDim = startFeature + ii * C10_WARP_SIZE; in C10_LAUNCH_BOUNDS_1() local232 int featureDim = startFeature + ii * C10_WARP_SIZE; in C10_LAUNCH_BOUNDS_1() local272 int featureDim = startFeature + ii * C10_WARP_SIZE; in coalesceValuesKernel() local282 int featureDim = startFeature + ii * C10_WARP_SIZE; in coalesceValuesKernel() local
77 int64_t featureDim = (chunk % chunksPerBag) * blockDim.x + threadIdx.x; in EmbeddingBag_updateOutputKernel_max() local129 int64_t featureDim = (chunk % chunksPerBag) * blockDim.x + threadIdx.x; in EmbeddingBag_updateOutputKernel_sum_mean() local253 int64_t featureDim = (chunk % chunksPerBag) * blockDim.x + threadIdx.x; in EmbeddingBag_accGradParametersKernel_max() local
40 Value value1d, Value shapeValue, int64_t featureDim, in broadcastToFeatureDim()120 int64_t featureDim = bnOp.feature_index(); in matchAndRewrite() local