forked from nudt_dsp/netrans
18 lines
460 B
C
18 lines
460 B
C
#ifndef TH_CUDA_TENSOR_RANDOM_INC
|
|
#define TH_CUDA_TENSOR_RANDOM_INC
|
|
|
|
#include <THC/THCTensor.h>
|
|
|
|
#include <THC/generic/THCTensorRandom.h>
|
|
#include <THC/THCGenerateAllTypes.h>
|
|
|
|
#include <THC/generic/THCTensorRandom.h>
|
|
#include <THC/THCGenerateBoolType.h>
|
|
|
|
#include <ATen/CUDAGenerator.h>
|
|
|
|
THC_API void THCRandom_getRNGState(at::Generator *gen_, THByteTensor *rng_state);
|
|
THC_API void THCRandom_setRNGState(at::Generator *gen_, THByteTensor *rng_state);
|
|
|
|
#endif
|