diff options
-rw-r--r-- | lib/TH/THStorage.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/lib/TH/THStorage.h b/lib/TH/THStorage.h index 7aab593..c8ab484 100644 --- a/lib/TH/THStorage.h +++ b/lib/TH/THStorage.h @@ -30,8 +30,11 @@ 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); + TH_API int THLongStorage_inferExpandGeometry(long *tensorSizes, long *tensorStrides, long tensorDim, THLongStorage *sizes, long **esz, long **est, int raiseErrors); #endif |