Searched refs:get_nested_strides (Results 1 – 8 of 8) sorted by relevance
56 self_ptr->get_nested_strides(), in get_elementwise_nested_tensor_impl()57 other_ptr->get_nested_strides()), in get_elementwise_nested_tensor_impl()89 other_impl->get_nested_strides().clone(), in NestedTensor_elementwise_Tensor()99 self_impl->get_nested_strides().clone(), in NestedTensor_elementwise_Tensor()178 self_impl->get_nested_strides(), in NestedTensor_elementwise_Tensor()
27 return get_nested_tensor_impl(self) -> get_nested_strides(); in _nested_tensor_strides()88 const auto& strides = self_impl->get_nested_strides(); in chunk_nested_tensor()143 const auto& strides = self_impl->get_nested_strides(); in split_with_sizes_nested()
64 auto nested_strides = self_nt->get_nested_strides().clone(); in empty_like_nested()141 stridemat = self_ptr->get_nested_strides(); in clone_nested()213 auto nested_strides = nt_impl->get_nested_strides(); in narrow_nested_symint()237 const auto& nested_strides = nt_impl->get_nested_strides(); in alias_nested()
176 nt_input->get_nested_strides(), in _pin_memory_nested()
487 & stridemat = input_ptr->get_nested_strides(); in native_dropout_nested()598 & stridemat = self_ptr->get_nested_strides(); in transpose_nested()627 const Tensor& stridemat = self_ptr->get_nested_strides(); in squeeze_dim_nested()674 const Tensor& stridemat = self_ptr->get_nested_strides(); in unsqueeze_nested()
148 const Tensor& stridemat = self_ptr->get_nested_strides(); in NestedTensor_get_strides()
55 const Tensor& get_nested_strides() const { in get_nested_strides() function230 &stridemat = nt->get_nested_strides(); in nested_tensor_impl_is_contiguous()
89 const Tensor& tensor_strides = tensor->get_nested_strides(); in is_safe_to_get_storage_as_tensor()182 tensor_impl->get_nested_strides().data_ptr<int64_t>(); in view_as_dense()