Home
last modified time | relevance | path

Searched refs:FunctionalTensorWrapper (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFunctionalTensorWrapper.cpp22 void FunctionalTensorWrapper::set_constructor_metadata() { in set_constructor_metadata()
51 FunctionalTensorWrapper::FunctionalTensorWrapper(const Tensor& value) in FunctionalTensorWrapper() function in at::FunctionalTensorWrapper
64 void FunctionalTensorWrapper::freeze_storage() const { in freeze_storage()
132 FunctionalTensorWrapper::FunctionalTensorWrapper(const Tensor& view_value, const FunctionalTensorWr… in FunctionalTensorWrapper() function in at::FunctionalTensorWrapper
156 functionalization::FunctionalStorageImpl* FunctionalTensorWrapper::functional_storage_impl() const { in functional_storage_impl()
160 void FunctionalTensorWrapper::commit_update() { in commit_update()
173 bool FunctionalTensorWrapper::is_up_to_date() const { in is_up_to_date()
179 void FunctionalTensorWrapper::mutate_view_meta(const at::functionalization::ViewMeta& meta) { in mutate_view_meta()
219 void FunctionalTensorWrapper::replace_(const Tensor& other, bool from_lazy_regenerate) { in replace_()
254 bool FunctionalTensorWrapper::has_data_mutation() { in has_data_mutation()
[all …]
H A DFunctionalTensorWrapper.h50 struct TORCH_API FunctionalTensorWrapper : public c10::TensorImpl { struct
51 explicit FunctionalTensorWrapper(const Tensor& value);
56 explicit FunctionalTensorWrapper(
58 const FunctionalTensorWrapper* base,
150 void set__impl(const FunctionalTensorWrapper* other);
220 ~FunctionalTensorWrapper() override = default;
254 const FunctionalTensorWrapper* src_impl,
255 FunctionalTensorWrapper* dest_impl,
280 const FunctionalTensorWrapper* src_impl, argument
281 FunctionalTensorWrapper* dest_impl,
[all …]
/aosp_15_r20/external/pytorch/
H A Dbuild_variables.bzl1010 "aten/src/ATen/FunctionalTensorWrapper.cpp",