Searched refs:nested_sizes_ (Results 1 – 3 of 3) sorted by relevance
173 nested_sizes_(std::move(nested_sizes)), in NestedTensorImpl()186 validate_nested_tensor_metadata(nested_sizes_, nested_strides_, storage_offsets_); in NestedTensorImpl()230 nested_sizes_(std::move(nested_sizes)), in NestedTensorImpl()234 validate_nested_tensor_metadata(nested_sizes_, nested_strides_, storage_offsets_); in NestedTensorImpl()242 opt_sizes_ = std::make_optional(construct_opt_sizes(nested_sizes_)); in opt_size()252 const auto my_dim = nested_sizes_.dim() ? nested_sizes_.sizes()[1] + 1 : 1; in refresh_dim()263 if (nested_sizes_.dim() == 0) { in numel_custom()266 return get_numel_from_nested_size_tensor(nested_sizes_); in numel_custom()316 nested_sizes_, in shallow_copy_and_detach_core()
52 return nested_sizes_; in get_nested_sizes()155 const at::Tensor nested_sizes_, nested_strides_; member
8 - nested_sizes_: 2d tensor of n_tensor_components x n_dims