/usr/local/lib64/python3.6/site-packages/torch/include/THC/generic
NameSizeModeActions
THCStorage.h26950644editdlrm
THCStorageCopy.h56460644editdlrm
THCTensor.h12850644editdlrm
THCTensorCopy.h5100644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/THC/generic/THCStorage.h (2695B)
#ifndef THC_GENERIC_FILE #define THC_GENERIC_FILE "THC/generic/THCStorage.h" #else #define THCStorage THStorage // These used to be distinct types; for some measure of backwards compatibility and documentation // alias these to the single THCStorage type. #define THCudaStorage THCStorage #define THCudaDoubleStorage THCStorage #define THCudaHalfStorage THCStorage #define THCudaByteStorage THCStorage #define THCudaCharStorage THCStorage #define THCudaShortStorage THCStorage #define THCudaIntStorage THCStorage #define THCudaLongStorage THCStorage #define THCudaBoolStorage THCStorage #define THCudaBFloat16Storage THCStorage #define THCudaComplexFloatStorage THCStorage #define THCudaComplexDoubleStorage THCStorage TORCH_CUDA_CU_API scalar_t* THCStorage_( data)(THCState* state, const THCStorage*); TORCH_CUDA_CU_API int THCStorage_(elementSize)(THCState* state); /* slow access -- checks everything */ TORCH_CUDA_CU_API void THCStorage_( set)(THCState* state, THCStorage*, ptrdiff_t, scalar_t); TORCH_CUDA_CU_API scalar_t THCStorage_(get)(THCState* state, const THCStorage*, ptrdiff_t); TORCH_CUDA_CU_API THCStorage* THCStorage_(new)(THCState* state); TORCH_CUDA_CU_API THCStorage* THCStorage_( newWithSize)(THCState* state, ptrdiff_t size); TORCH_CUDA_CU_API THCStorage* THCStorage_( newWithSize1)(THCState* state, scalar_t); TORCH_CUDA_CU_API THCStorage* THCStorage_(newWithMapping)( THCState* state, const char* filename, ptrdiff_t size, int shared); TORCH_CUDA_CU_API THCStorage* THCStorage_(newWithAllocator)( THCState* state, ptrdiff_t size, at::Allocator* allocator); TORCH_CUDA_CU_API THCStorage* THCStorage_(newWithDataAndAllocator)( THCState* state, at::DataPtr&& data, ptrdiff_t size, at::Allocator* allocator); TORCH_CUDA_CU_API void THCStorage_( setFlag)(THCState* state, THCStorage* storage, const char flag); TORCH_CUDA_CU_API void THCStorage_( clearFlag)(THCState* state, THCStorage* storage, const char flag); TORCH_CUDA_CU_API void THCStorage_( retain)(THCState* state, THCStorage* storage); TORCH_CUDA_CU_API void THCStorage_(free)(THCState* state, THCStorage* storage); TORCH_CUDA_CU_API void THCStorage_( resizeBytes)(THCState* state, THCStorage* storage, ptrdiff_t size_bytes); TORCH_CUDA_CU_API void THCStorage_( fill)(THCState* state, THCStorage* storage, scalar_t value); TORCH_CUDA_CU_API int THCStorage_( getDevice)(THCState* state, const THCStorage* storage); #endif