Searched defs:input_target_k (Results 1 – 2 of 2) sorted by relevance
101 scalar_t input_target_k = input_k[target_idx]; in C10_LAUNCH_BOUNDS_1() local171 scalar_t input_target_k = input_k[target_idx]; in C10_LAUNCH_BOUNDS_1() local
34 scalar_t input_target_k = input_k[target_k]; in MultiMarginLoss_forward_kernel() local79 scalar_t input_target_k = input_k[target_k]; in MultiMarginLoss_backward_kernel() local