Lines Matching defs:NestedTensorImpl
20 struct TORCH_API NestedTensorImpl : public c10::TensorImpl { struct
51 const Tensor& get_nested_sizes() const { in get_nested_sizes()
55 const Tensor& get_nested_strides() const { in get_nested_strides()
58 const Tensor& get_storage_offsets() const { in get_storage_offsets()
66 int64_t size(int64_t d) const { in size()
83 at::Tensor get_buffer() const { in get_buffer()
97 at::Tensor get_unsafe_storage_as_tensor() const { in get_unsafe_storage_as_tensor()
107 size_t get_buffer_size() const { in get_buffer_size()
119 int64_t size_custom(int64_t d) const override { in size_custom()
122 c10::SymInt sym_size_custom(int64_t d) const override { in sym_size_custom()
141 void shallow_copy_from(const c10::intrusive_ptr<TensorImpl>& impl) override { in shallow_copy_from()
155 const at::Tensor nested_sizes_, nested_strides_;
169 const at::Tensor storage_offsets_;
174 mutable std::optional<std::vector<int64_t>> opt_sizes_;
210 inline NestedTensorImpl* get_nested_tensor_impl_or_null( in get_nested_tensor_impl_or_null() argument