Home
last modified time | relevance | path

Searched refs:EIGEN_HAS_CUDA_FP16 (Results 1 – 9 of 9) sorted by relevance

/external/eigen/Eigen/src/Core/arch/CUDA/
DTypeCasting.h22 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
40 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
58 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
72 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
DHalf.h52 #if !defined(EIGEN_HAS_CUDA_FP16)
77 #if !defined(EIGEN_HAS_CUDA_FP16)
143 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530
284 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 in float_to_half_rtne()
339 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 in half_to_float()
373 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530
392 #if defined(EIGEN_HAS_CUDA_FP16) && defined __CUDACC_VER__ && __CUDACC_VER__ >= 80000 && defined(__… in log()
430 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530 in half()
439 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530 in half()
DPacketMathHalf.h18 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDACC__) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ …
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorReductionCuda.h82 #ifdef EIGEN_HAS_CUDA_FP16
188 #ifdef EIGEN_HAS_CUDA_FP16
305 #ifdef EIGEN_HAS_CUDA_FP16
347 #ifdef EIGEN_HAS_CUDA_FP16
442 #ifdef EIGEN_HAS_CUDA_FP16
585 #ifdef EIGEN_HAS_CUDA_FP16
637 #ifdef EIGEN_HAS_CUDA_FP16
DTensorMeta.h55 #if defined(EIGEN_USE_GPU) && defined(__CUDACC__) && defined(EIGEN_HAS_CUDA_FP16)
DTensorReduction.h329 #ifdef EIGEN_HAS_CUDA_FP16
684 #ifdef EIGEN_HAS_CUDA_FP16
/external/eigen/unsupported/test/
Dcxx11_tensor_of_float16_cuda.cu62 #ifdef EIGEN_HAS_CUDA_FP16
482 #ifdef EIGEN_HAS_CUDA_FP16 in test_cxx11_tensor_of_float16_cuda()
/external/eigen/Eigen/
DCore233 #define EIGEN_HAS_CUDA_FP16
237 #if defined EIGEN_HAS_CUDA_FP16
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc44 #define EIGEN_HAS_CUDA_FP16 macro