Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp2410 int64_t total_numel = 0; in checkForNCCLErrorsInternal() local
3487 auto total_numel = check_gpu_tensors_same_device(tensors); in checkForNCCLErrorsInternal() local
4263 int64_t total_numel = 0; in checkForNCCLErrorsInternal() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh556 int64_t total_numel = 0; in batch_norm_backward_elemt_kernel() local
1392 int64_t total_numel = 0; in batch_norm_backward_elemt_channels_last_kernel() local