Searched refs:num_elements_nested_size (Results 1 – 1 of 1) sorted by relevance
882 uint64_t num_elements_nested_size = at::prod(nested_sizes, 1).sum().item<int64_t>(); in _nested_view_from_buffer() local885 buffer_storage_size == num_elements_nested_size, in _nested_view_from_buffer()889 num_elements_nested_size, in _nested_view_from_buffer()