Home
last modified time | relevance | path

Searched refs:__macro (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver_wrapper.h73 #define HIP_ROUTINE_EACH(__macro) \ argument
74 __macro(hipDeviceCanAccessPeer) \
75 __macro(hipDeviceEnablePeerAccess) \
76 __macro(hipDeviceGet) \
77 __macro(hipDeviceGetAttribute) \
78 __macro(hipDeviceGetName) \
79 __macro(hipDeviceGetPCIBusId) \
80 __macro(hipDeviceGetSharedMemConfig) \
81 __macro(hipDeviceSetSharedMemConfig) \
82 __macro(hipDeviceSynchronize) \
[all …]
Drocm_fft.cc89 #define ROCFFT_ROUTINE_EACH(__macro) \ argument
90 __macro(hipfftDestroy) __macro(hipfftSetStream) __macro(hipfftPlan1d) \
91 __macro(hipfftPlan2d) __macro(hipfftPlan3d) __macro(hipfftPlanMany) \
92 __macro(hipfftCreate) __macro(hipfftSetAutoAllocation) \
93 __macro(hipfftSetWorkArea) __macro(hipfftGetSize1d) \
94 __macro(hipfftMakePlan1d) __macro(hipfftGetSize2d) \
95 __macro(hipfftMakePlan2d) __macro(hipfftGetSize3d) \
96 __macro(hipfftMakePlan3d) __macro(hipfftGetSizeMany) \
97 __macro(hipfftMakePlanMany) \
98 __macro(hipfftExecD2Z) \
[all …]
Drocm_dnn.cc160 #define MIOPEN_DNN_ROUTINE_EACH(__macro) \ argument
161 __macro(miopenBatchNormalizationBackward) \
162 __macro(miopenBatchNormalizationForwardInference) \
163 __macro(miopenBatchNormalizationForwardTraining) \
164 __macro(miopenGetConvolutionForwardOutputDim) \
165 __macro(miopenFindConvolutionForwardAlgorithm) \
166 __macro(miopenCreateTensorDescriptor) \
167 __macro(miopenDestroyTensorDescriptor) \
168 __macro(miopenSet2dPoolingDescriptor) \
169 __macro(miopenSetLRNDescriptor) \
[all …]
Drocm_blas.cc103 #define ROCBLAS_BLAS_ROUTINE_EACH(__macro) \ argument
104 __macro(rocblas_snrm2) __macro(rocblas_dnrm2) /* __macro(rocblas_scnrm2) \
106 __macro(rocblas_sdot) \
107 __macro(rocblas_ddot) /* __macro(rocblas_cdotu) \
111 __macro(rocblas_sscal) \
112 __macro(rocblas_dscal) /* __macro(rocblas_cscal) \
116 __macro(rocblas_saxpy) \
117 __macro(rocblas_daxpy) /* __macro(rocblas_caxpy) \
119 __macro(rocblas_scopy) \
120 __macro(rocblas_dcopy) /* __macro(rocblas_ccopy) \
[all …]