Searched defs:output_k (Results 1 – 3 of 3) sorted by relevance
72 int output_k = k - c; in TEST() local135 int output_k = k - c; in TEST() local207 int output_k = k - c + forward_pad_x; in TEST() local278 int output_k = k - c + forward_pad_x; in TEST() local344 int output_k = k - c; in TEST() local411 int output_k = k - c; in TEST() local477 int output_k = k - c; in TEST() local543 int output_k = k - c; in TEST() local609 int output_k = k - c; in TEST() local676 int output_k = k - c; in TEST() local[all …]
70 scalar_t* output_k = output + k; in C10_LAUNCH_BOUNDS_1() local
31 scalar_t *output_k = output + k; in MultiMarginLoss_forward_kernel() local