Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/
DEmptyTensor.h28 TORCH_API size_t computeStorageNbytesContiguous(
32 TORCH_API SymInt computeStorageNbytesContiguous(
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()
DFunctionalizeFallbackKernel.cpp153 auto new_size_bytes = at::detail::computeStorageNbytesContiguous(size, itemsize, storage_offset); in resize__functionalization()
/external/pytorch/aten/src/ATen/native/cuda/
DResize.h45 storage_size = at::detail::computeStorageNbytesContiguous( in resize_impl_cuda_()
/external/pytorch/aten/src/ATen/native/
DResize.cpp215 storage_size = at::detail::computeStorageNbytesContiguous( in _resize_impl_()
DTensorShape.cpp424 ? at::detail::computeStorageNbytesContiguous(size, itemsize, std::move(storage_offset)) in set_storage_meta__symint()
/external/pytorch/test/cpp_extensions/
Dopen_registration_extension.cpp427 const auto storage_size = at::detail::computeStorageNbytesContiguous(size, itemsize, offset); in custom_resize_()
/external/pytorch/torch/csrc/autograd/
Dpython_variable.cpp834 size_bytes = at::detail::computeStorageNbytesContiguous( in THPVariable_make_wrapper_subclass()