Searched defs:map_nelem (Results 1 – 1 of 1) sorted by relevance
312 int64_t map_nelem = target.numel() / batch_size; in nll_loss2d_forward_out_cuda_template() local435 int64_t map_nelem = target_numel / batch_size; in nll_loss2d_backward_out_cuda_template() local