Home
last modified time | relevance | path

Searched refs:computeStorageNbytesContiguous (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DEmptyTensor.h28 TORCH_API size_t computeStorageNbytesContiguous(
32 TORCH_API SymInt computeStorageNbytesContiguous(
H A DEmptyTensor.cpp53 size_t computeStorageNbytesContiguous( in computeStorageNbytesContiguous() function
124 SymInt computeStorageNbytesContiguous( in computeStorageNbytesContiguous() function
172 auto size_bytes = computeStorageNbytesContiguous(size, dtype.itemsize()); in _empty_generic()
H A DFunctionalizeFallbackKernel.cpp153 auto new_size_bytes = at::detail::computeStorageNbytesContiguous(size, itemsize, storage_offset); in resize__functionalization()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DResize.h45 storage_size = at::detail::computeStorageNbytesContiguous( in resize_impl_cuda_()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DResize.cpp215 storage_size = at::detail::computeStorageNbytesContiguous( in _resize_impl_()
H A DTensorShape.cpp424 ? at::detail::computeStorageNbytesContiguous(size, itemsize, std::move(storage_offset)) in set_storage_meta__symint()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp427 const auto storage_size = at::detail::computeStorageNbytesContiguous(size, itemsize, offset); in custom_resize_()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable.cpp834 size_bytes = at::detail::computeStorageNbytesContiguous( in THPVariable_make_wrapper_subclass()