Searched refs:create_nested_view_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorMatmul.cpp | 144 auto viewed_self = create_nested_view_tensor( in matmul_with_bmm_nested()
148 auto viewed_mat2 = create_nested_view_tensor( in matmul_with_bmm_nested()
179 auto viewed_out = create_nested_view_tensor( in matmul_with_bmm_nested()
|
| H A D | NestedTensorUtils.cpp | 108 splits[split_idx] = create_nested_view_tensor(self, new_sizes, new_strides, new_offsets); in chunk_nested_tensor()
164 splits[split_idx] = create_nested_view_tensor(self, new_sizes, new_strides, new_offsets); in split_with_sizes_nested()
|
| H A D | NestedTensorMath.cpp | 478 return create_nested_view_tensor(self, new_sizes, new_strides, new_offsets); in select_nested()
607 return create_nested_view_tensor( in transpose_nested()
658 return create_nested_view_tensor( in squeeze_dim_nested()
689 return create_nested_view_tensor( in unsqueeze_nested()
846 …return create_nested_view_tensor(self, sizemat_reshaped, stridemat_reshaped, self_ptr->get_storage… in view_nested()
|
| H A D | NestedTensorUtils.h | 82 inline at::Tensor create_nested_view_tensor( in create_nested_view_tensor() function
|