/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
torch
/
include
/
THC
/
generic
/
/usr/local/lib64/python3.6/site-packages/torch/include/THC/generic
mkdir
upload
Name
Size
Mode
Actions
THCStorage.h
2695
0644
edit
dl
rm
THCStorageCopy.h
5646
0644
edit
dl
rm
THCTensor.h
1285
0644
edit
dl
rm
THCTensorCopy.h
510
0644
edit
dl
rm
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
Save
cmd:
run