Searched defs:new_numel (Results 1 – 3 of 3) sorted by relevance
104 const auto new_numel = compute_numel(new_sizes.data(), dim_); in internal_resize_contiguous() local
398 const int64_t new_numel = at::sum(num_segments).item<int64_t>(); in NestedTensor_sum_dim_CPU() local
2473 int64_t new_numel = 1; in SetDimsTemplate() local