diff options
Diffstat (limited to 'lib/TH/THStorage.h')
-rw-r--r-- | lib/TH/THStorage.h | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/lib/TH/THStorage.h b/lib/TH/THStorage.h index 0b61ff8..e141e4c 100644 --- a/lib/TH/THStorage.h +++ b/lib/TH/THStorage.h @@ -31,12 +31,14 @@ typedef struct { TH_API THDescBuff THLongStorage_sizeDesc(const THLongStorage *size); TH_API THLongStorage *THLongStorage_newInferSize(THLongStorage *size, ptrdiff_t nElement); -// Given the sizes of {2,N} tensors, write out the size when the tensors are expanded together -TH_API int THLongStorage_inferSize2(THLongStorage *output, long *sizesA, long dimsA, long *sizesB, long dimsB, int raiseErrors); -TH_API int THLongStorage_inferSizeN(THLongStorage *output, int n, long **sizes, long *dims, int raiseErrors); +// Given the sizes of {2,N} tensors, write out the size when the tensors are expanded together. +TH_API int THLongStorage_inferSize2(THLongStorage *output, long *sizesA, long dimsA, + long *sizesB, long dimsB, char *error_buffer, int buffer_len); +TH_API int THLongStorage_inferSizeN(THLongStorage *output, int n, long **sizes, long *dims, + char *error_buffer, int buffer_len); TH_API int THLongStorage_inferExpandGeometry(long *tensorSizes, long *tensorStrides, long tensorDim, THLongStorage *sizes, long **expandedSizes, long **expandedStrides, - int raiseErrors); + char *error_buffer, int buffer_len); #endif |