Searched defs:target_k (Results 1 – 2 of 2) sorted by relevance
69 const int64_t* target_k = target + k * dim; in C10_LAUNCH_BOUNDS_1() local143 const int64_t* target_k = target + k * dim; in C10_LAUNCH_BOUNDS_1() local
32 int target_k = static_cast<int>(target[k]); in MultiMarginLoss_forward_kernel() local78 int target_k = static_cast<int>(target[k]); in MultiMarginLoss_backward_kernel() local