| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | NestedTensorImpl.cpp | 165 NestedTensorImpl::NestedTensorImpl( in NestedTensorImpl() function in at::native::NestedTensorImpl
191 NestedTensorImpl::NestedTensorImpl( in NestedTensorImpl() function in at::native::NestedTensorImpl
196 : NestedTensorImpl( in NestedTensorImpl()
213 NestedTensorImpl::NestedTensorImpl( in NestedTensorImpl() function in at::native::NestedTensorImpl
216 : NestedTensorImpl( in NestedTensorImpl()
223 NestedTensorImpl::NestedTensorImpl( in NestedTensorImpl() function in at::native::NestedTensorImpl
239 std::optional<int64_t> NestedTensorImpl::opt_size(int64_t d) const { in opt_size()
251 void NestedTensorImpl::refresh_dim() { in refresh_dim()
257 int64_t NestedTensorImpl::dim_custom() const { in dim_custom()
262 int64_t NestedTensorImpl::numel_custom() const { in numel_custom()
[all …]
|
| H A D | NestedTensorImpl.h | 14 struct NestedTensorImpl;
15 inline bool nested_tensor_impl_is_contiguous(const NestedTensorImpl* nt);
20 struct TORCH_API NestedTensorImpl : public c10::TensorImpl { struct
21 explicit NestedTensorImpl(
29 explicit NestedTensorImpl(
36 explicit NestedTensorImpl(
41 explicit NestedTensorImpl(
210 inline NestedTensorImpl* get_nested_tensor_impl_or_null( in get_nested_tensor_impl_or_null() argument
213 return static_cast<NestedTensorImpl*>(tensor.unsafeGetTensorImpl()); in get_nested_tensor_impl_or_null()
218 inline NestedTensorImpl* get_nested_tensor_impl(const at::Tensor& tensor) { in get_nested_tensor_impl()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorUtils.h | 29 struct NestedTensorImpl;
42 return at::detail::make_tensor<NestedTensorImpl>( in wrap_buffer()
54 return at::detail::make_tensor<NestedTensorImpl>( in wrap_buffer()
94 return at::detail::make_tensor<NestedTensorImpl>( in create_nested_view_tensor()
105 int64_t get_consistent_last_dim_of_nested_tensor(const NestedTensorImpl& nt);
109 const NestedTensorImpl* self_ptr) { in NestedTensor_get_sizes()
131 const NestedTensorImpl& nt);
137 const NestedTensorImpl* self_ptr = get_nested_tensor_impl(self); in NestedTensor_get_sizes()
142 const NestedTensorImpl* self_ptr) { in NestedTensor_get_strides()
164 const NestedTensorImpl* self_ptr = get_nested_tensor_impl(self); in NestedTensor_get_strides()
[all …]
|
| H A D | README.md | 13 …m c10::TensorImpl whose definition can be found here: [NestedTensorImpl.h](../../NestedTensorImpl.…
15 …NestedTensorImpl constructor directly but using the `wrap_buffer` function defined in [NestedTenso…
21 - `NestedTensorImpl.h | NestedTensorImpl.cpp`: The NestedTensor data structure and its methods.
41 return at::detail::make_tensor<NestedTensorImpl>(f(nt_impl->get_buffer()), sizes);
44 1. Get the NestedTensorImpl from the input NestedTensor.
|
| H A D | NestedTensorMath.h | 18 return at::detail::make_tensor<NestedTensorImpl>(f(nt_impl->get_buffer()), sizes); in map_nt()
25 …return at::detail::make_tensor<NestedTensorImpl>(f(nt_impl_1->get_buffer(), nt_impl_2->get_buffer(… in map_nt_binary()
29 const NestedTensorImpl& input, in _check_nested_layer_norm_inputs()
|
| H A D | NestedTensorTransformerFunctions.h | 21 struct NestedTensorImpl;
51 const NestedTensorImpl& nt);
|
| H A D | NestedTensorUtils.cpp | 55 std::vector<int64_t> NestedTensor_get_max_size(const NestedTensorImpl& nt) { in NestedTensor_get_max_size()
60 int64_t get_consistent_last_dim_of_nested_tensor(const NestedTensorImpl& nt) { in get_consistent_last_dim_of_nested_tensor()
|
| H A D | NestedTensorMath.cpp | 241 return at::detail::make_tensor<NestedTensorImpl>( in nested_from_padded_generic()
351 return at::detail::make_tensor<NestedTensorImpl>( in NestedTensor_embedding()
895 return at::detail::make_tensor<NestedTensorImpl>( in _nested_view_from_buffer()
1057 return at::detail::make_tensor<NestedTensorImpl>( in cat_nested_as_jagged()
1080 return at::detail::make_tensor<NestedTensorImpl>( in cat_nested_impl()
|
| H A D | NestedTensorFactories.cpp | 225 return at::detail::make_tensor<NestedTensorImpl>( in narrow_nested_symint()
239 return at::detail::make_tensor<NestedTensorImpl>( in alias_nested()
|
| H A D | NestedTensorBinaryOps.cpp | 22 std::pair<NestedTensorImpl*, NestedTensorImpl*>
|
| H A D | NestedTensorMatmul.cpp | 204 return at::detail::make_tensor<NestedTensorImpl>( in matmul_nested_with_broadcasted_dense()
|
| H A D | NestedTensorTransformerFunctions.cpp | 120 return at::detail::make_tensor<NestedTensorImpl>( in NestedTensor_times_Tensor_plus_Tensor_addmm()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorTransformerFunctions.cpp | 116 return at::detail::make_tensor<NestedTensorImpl>(std::move(output), sizes); in nested_from_padded_cuda()
159 Tensor result = at::detail::make_tensor<NestedTensorImpl>( in NestedTensor_to_padded_tensor_cuda()
|
| H A D | NestedTensorTransformerUtils.cpp | 85 bool is_safe_to_get_storage_as_tensor(const NestedTensorImpl* tensor) { in is_safe_to_get_storage_as_tensor()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/ |
| H A D | transformer.cpp | 100 …? at::detail::make_tensor<NestedTensorImpl>(check_for_empty, get_nested_tensor_impl(src)->get_nest… in transformer_encoder_layer_forward()
|
| /aosp_15_r20/external/pytorch/ |
| H A D | build_variables.bzl | 1015 "aten/src/ATen/NestedTensorImpl.cpp",
|