diff options
Diffstat (limited to 'lib/TH/THStorage.h')
| -rw-r--r-- | lib/TH/THStorage.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/lib/TH/THStorage.h b/lib/TH/THStorage.h index c8ab484..0b61ff8 100644 --- a/lib/TH/THStorage.h +++ b/lib/TH/THStorage.h @@ -35,6 +35,8 @@ TH_API THLongStorage *THLongStorage_newInferSize(THLongStorage *size, ptrdiff_t 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); +TH_API int THLongStorage_inferExpandGeometry(long *tensorSizes, long *tensorStrides, long tensorDim, + THLongStorage *sizes, long **expandedSizes, long **expandedStrides, + int raiseErrors); #endif |