forked from nudt_dsp/netrans
27 lines
540 B
C
27 lines
540 B
C
#ifndef THC_TENSOR_INC
|
|
#define THC_TENSOR_INC
|
|
|
|
#include <TH/THTensor.h>
|
|
#include <THC/THCStorage.h>
|
|
#include <THC/THCGeneral.h>
|
|
|
|
#define THCTensor_(NAME) TH_CONCAT_4(TH,CReal,Tensor_,NAME)
|
|
|
|
#define THC_DESC_BUFF_LEN 64
|
|
|
|
typedef struct THC_CLASS THCDescBuff
|
|
{
|
|
char str[THC_DESC_BUFF_LEN];
|
|
} THCDescBuff;
|
|
|
|
#include <THC/generic/THCTensor.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensor.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensor.h>
|
|
#include <THC/THCGenerateBFloat16Type.h>
|
|
|
|
#endif
|