Home
last modified time | relevance | path

Searched refs:self_impl (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/aten/src/ATen/native/nested/
DNestedTensorBinaryOps.cpp95 auto self_impl = get_nested_tensor_impl(self); in NestedTensor_elementwise_Tensor() local
97 f(self_impl->get_unsafe_storage_as_tensor(), other), in NestedTensor_elementwise_Tensor()
98 self_impl->get_nested_sizes().clone(), in NestedTensor_elementwise_Tensor()
99 self_impl->get_nested_strides().clone(), in NestedTensor_elementwise_Tensor()
100 self_impl->get_storage_offsets() in NestedTensor_elementwise_Tensor()
170 auto [self_impl, other_impl] = in NestedTensor_elementwise_Tensor()
172 TORCH_INTERNAL_ASSERT_DEBUG_ONLY(self_impl); in NestedTensor_elementwise_Tensor()
175 f(self_impl->get_unsafe_storage_as_tensor(), in NestedTensor_elementwise_Tensor()
177 self_impl->get_nested_sizes(), in NestedTensor_elementwise_Tensor()
178 self_impl->get_nested_strides(), in NestedTensor_elementwise_Tensor()
[all …]
DNestedTensorUtils.cpp79 auto self_impl = get_nested_tensor_impl(self); in chunk_nested_tensor() local
80 const int64_t last_dim_size = get_consistent_last_dim_of_nested_tensor(*self_impl); in chunk_nested_tensor()
87 const auto& sizes = self_impl->get_nested_sizes(); in chunk_nested_tensor()
88 const auto& strides = self_impl->get_nested_strides(); in chunk_nested_tensor()
89 const auto offsets = self_impl->get_storage_offsets(); in chunk_nested_tensor()
134 auto self_impl = get_nested_tensor_impl(self); in split_with_sizes_nested() local
135 auto self_size = get_consistent_last_dim_of_nested_tensor(*self_impl); in split_with_sizes_nested()
142 const auto& sizes = self_impl->get_nested_sizes(); in split_with_sizes_nested()
143 const auto& strides = self_impl->get_nested_strides(); in split_with_sizes_nested()
144 const auto offsets = self_impl->get_storage_offsets(); in split_with_sizes_nested()
DNestedTensorUtils.h169 auto self_impl = get_nested_tensor_impl(self); in check_numel_equals_buffer_size() local
171 self.numel() == static_cast<int64_t>(self_impl->get_buffer_size()), in check_numel_equals_buffer_size()
/external/pytorch/torch/csrc/autograd/
Dvariable.h266 void set_requires_grad(bool requires_grad, at::TensorImpl* self_impl) final { in set_requires_grad()
269 isDifferentiableType(at::typeMetaToScalarType(self_impl->dtype())), in set_requires_grad()
297 at::TensorImpl* self_impl = nullptr,
305 TORCH_INTERNAL_ASSERT(self_impl);
306 set_requires_grad(requires_grad, self_impl);
768 at::TensorImpl* self_impl,
Dvariable.cpp43 at::TensorImpl* self_impl, in DifferentiableViewMeta() argument
48 : AutogradMeta(self_impl), in DifferentiableViewMeta()
55 self_impl->set_version_counter( in DifferentiableViewMeta()
57 attr_version_ = self_impl->version_counter().current_version(); in DifferentiableViewMeta()
59 backward_info_.value().base_.unsafeGetTensorImpl() != self_impl); in DifferentiableViewMeta()
/external/pytorch/aten/src/ATen/
DLegacyBatchedFallback.cpp103 const auto* self_impl = maybeGetBatchedImpl(self); in batchedTensorInplaceForLoopFallback() local
105 if (self_impl) { in batchedTensorInplaceForLoopFallback()
106 self_vmap_levels = createVmapLevelsBitset(self_impl->bdims()); in batchedTensorInplaceForLoopFallback()
DFunctionalizeFallbackKernel.cpp340 auto self_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(self); in set__functionalize() local
343 TORCH_CHECK(!self_impl->was_inductor_storage_resized(), in set__functionalize()
345 self_impl->set__impl(src_impl); in set__functionalize()
DFunctionalTensorWrapper.cpp786 auto self_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(self); in mutate_view_meta() local
787 self_impl->mutate_view_meta(meta); in mutate_view_meta()
/external/pytorch/aten/src/ATen/functorch/
DBatchedFallback.cpp108 const auto* self_impl = maybeGetBatchedImpl(self); in batchedTensorInplaceForLoopFallback() local
110 if (self_impl) { in batchedTensorInplaceForLoopFallback()
111 self_vmap_levels = createVmapLevelsBitset(self_impl->level()); in batchedTensorInplaceForLoopFallback()
/external/pytorch/c10/core/
DTensorImpl.h157 at::TensorImpl* self_impl) = 0;