#pragma once // STOP!!! Thinking of including this header directly? Please // read Note [TH abstraction violation] #include #include #include #include #include #include // See [NOTE: nDimension vs nDimensionLegacyNoScalars vs nDimensionLegacyAll] THC_API int THCTensor_nDimension(THCState *state, const THCTensor *self); THC_API int THCTensor_nDimensionLegacyNoScalars(THCState *state, const THCTensor *self); THC_API int THCTensor_nDimensionLegacyAll(THCState *state, const THCTensor *self); THC_API int64_t THCTensor_size(THCState *state, const THCTensor *self, int dim); THC_API int64_t THCTensor_sizeLegacyNoScalars(THCState *state, const THCTensor *self, int dim); THC_API int64_t THCTensor_stride(THCState *state, const THCTensor *self, int dim); THC_API int64_t THCTensor_strideLegacyNoScalars(THCState *state, const THCTensor *self, int dim); THC_API THCTensor *THCTensor_new(THCState *state, caffe2::TypeMeta type_meta); THC_API void THCTensor_resize(THCState *state, THCTensor *tensor, at::IntArrayRef size, at::IntArrayRef stride); THC_API void THCTensor_resizeNd(THCState *state, THCTensor *tensor, int nDimension, const int64_t *size, const int64_t *stride); THC_API void THCTensor_resizeAs(THCState *state, THCTensor *tensor, THCTensor *src); THC_API void THCTensor_set(THCState *state, THCTensor *self, THCTensor *src); THC_API void THCTensor_setStorage(THCState *state, THCTensor *self, THCStorage *storage_, ptrdiff_t storageOffset_, at::IntArrayRef size_, at::IntArrayRef stride_); THC_API void THCTensor_setStorageNd(THCState *state, THCTensor *self, THCStorage *storage, ptrdiff_t storageOffset, int nDimension, const int64_t *size, const int64_t *stride); THC_API void THCTensor_squeeze1d(THCState *state, THCTensor *self, THCTensor *src, int dimension_); THC_API void THCTensor_unsqueeze1d(THCState *state, THCTensor *self, THCTensor *src, int dimension_); THC_API bool THCTensor_allContiguous(THCState *state, THCTensor **inputs, int numInputs); THC_API ptrdiff_t THCTensor_nElement(THCState *state, const THCTensor *self); THC_API void THCTensor_retain(THCState *state, THCTensor *self); THC_API void THCTensor_free(THCState *state, THCTensor *self); THC_API int THCTensor_getDevice(THCState* state, const THCTensor* tensor); THC_API bool THCTensor_allSameDevice(THCState* state, THCTensor ** inputs, int numInputs); /* Can we use 32 bit math for indexing? */ THC_API bool THCTensor_canUse32BitIndexMath(THCState* state, const THCTensor* t, ptrdiff_t max_elem=INT32_MAX); /* Are all tensors 32-bit indexable? */ THC_API bool THCTensor_all32BitIndexable(THCState* state, THCTensor** inputs, int numInputs); THC_API void THCTensor_preserveReduceDimSemantics(THCState *state, THCTensor *tensor, int in_dims, int64_t dimension, int keepdim); /* Returns false if there is no possibility that the tensor */ /* has more than one index that references the same datapoint, */ /* true otherwise. */ THC_API bool THCTensor_maybeOverlappingIndices(THCState* state, const THCTensor* t); #include #include #include #include #include #include