forked from nudt_dsp/netrans
19 lines
387 B
C
19 lines
387 B
C
#ifndef TH_CUDA_TENSOR_COPY_INC
|
|
#define TH_CUDA_TENSOR_COPY_INC
|
|
|
|
#include <THC/THCTensor.h>
|
|
#include <THC/THCGeneral.h>
|
|
#include <TH/THHalf.h>
|
|
|
|
|
|
#include <THC/generic/THCTensorCopy.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorCopy.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorCopy.h>
|
|
#include <THC/THCGenerateBFloat16Type.h>
|
|
|
|
#endif
|