Searched defs:NestedTensorImpl (Results 1 – 2 of 2) sorted by relevance
20 struct TORCH_API NestedTensorImpl : public c10::TensorImpl { struct51 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()[all …]
165 NestedTensorImpl::NestedTensorImpl( in NestedTensorImpl() function in at::native::NestedTensorImpl191 NestedTensorImpl::NestedTensorImpl( in NestedTensorImpl() function in at::native::NestedTensorImpl213 NestedTensorImpl::NestedTensorImpl( in NestedTensorImpl() function in at::native::NestedTensorImpl223 NestedTensorImpl::NestedTensorImpl( in NestedTensorImpl() function in at::native::NestedTensorImpl