Home
last modified time | relevance | path

Searched defs:is_target_k (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DMultiLabelMarginCriterion.cu71 scalar_t* is_target_k = is_target + k * dim; in C10_LAUNCH_BOUNDS_1() local
144 const scalar_t* is_target_k = is_target + k * dim; in C10_LAUNCH_BOUNDS_1() local