Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DNormalization.cuh1043 __shared__ bool is_last_block_done; in batch_norm_collect_statistics_channels_last_kernel() local
1267 __shared__ bool is_last_block_done; in batch_norm_backward_reduce_channels_last_kernel() local
DReduce.cuh807 bool is_last_block_done = mark_block_finished(); in global_reduce() local