forked from nudt_dsp/netrans
86 lines
2.2 KiB
C
86 lines
2.2 KiB
C
#ifndef TH_CUDA_TENSOR_MATH_INC
|
|
#define TH_CUDA_TENSOR_MATH_INC
|
|
|
|
#include <THC/THCTensor.h>
|
|
#include <THC/THCGeneral.h>
|
|
|
|
#include <THC/generic/THCTensorMath.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMath.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorMath.h>
|
|
#include <THC/THCGenerateBFloat16Type.h>
|
|
|
|
#include <THC/generic/THCTensorMathBlas.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMathBlas.h>
|
|
#include <THC/THCGenerateBFloat16Type.h>
|
|
|
|
#include <THC/generic/THCTensorMathMagma.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMathPairwise.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMathPairwise.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorMathPointwise.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMathPointwise.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorMathReduce.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMathReduce.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorMathReduce.h>
|
|
#include <THC/THCGenerateBFloat16Type.h>
|
|
|
|
#include <THC/generic/THCTensorMathScan.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMathScan.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorMasked.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMasked.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorScatterGather.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorScatterGather.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorIndex.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorIndex.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <THC/generic/THCTensorIndex.h>
|
|
#include <THC/THCGenerateBFloat16Type.h>
|
|
|
|
#include <THC/generic/THCTensorMasked.h>
|
|
#include <THC/THCGenerateBFloat16Type.h>
|
|
|
|
#include <THC/generic/THCTensorSort.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorMode.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorTopK.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#endif
|