Home
last modified time | relevance | path

Searched defs:GPUDataType (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dmat_mul_op.cc743 struct GPUDataType<Eigen::half> { struct
745 static constexpr cudaDataType_t type = CUDA_R_16F;
750 struct GPUDataType<float> { struct
752 static constexpr cudaDataType_t type = CUDA_R_32F;
757 struct GPUDataType<std::complex<float>> { struct
759 static constexpr cudaDataType_t type = CUDA_C_32F;
764 struct GPUDataType<double> { struct
766 static constexpr cudaDataType_t type = CUDA_R_64F;
771 struct GPUDataType<std::complex<double>> { struct
773 static constexpr cudaDataType_t type = CUDA_C_64F;