mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
21 lines
537 B
C
21 lines
537 B
C
#ifndef THDP_UTILS_H
|
|
#define THDP_UTILS_H
|
|
|
|
#include "THDP.h"
|
|
|
|
#define THDPUtils_(NAME) TH_CONCAT_4(THDP,Real,Utils_,NAME)
|
|
|
|
#define THDStoragePtr TH_CONCAT_3(THD,Real,StoragePtr)
|
|
#define THDTensorPtr TH_CONCAT_3(THD,Real,TensorPtr)
|
|
#define THDPStoragePtr TH_CONCAT_3(THDP,Real,StoragePtr)
|
|
#define THDPTensorPtr TH_CONCAT_3(THDP,Real,TensorPtr)
|
|
|
|
#include "override_macros.h"
|
|
|
|
#define THD_GENERIC_FILE "torch/csrc/generic/utils.h"
|
|
#include <THD/base/THDGenerateAllTypes.h>
|
|
|
|
typedef THPPointer<THDTensorDescriptor> THDPTensorDesc;
|
|
|
|
#endif
|