Home
last modified time | relevance | path

Searched refs:GpuComplexT (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_helpers.h66 struct GpuComplexT { struct
71 struct GpuComplexT<std::complex<float>> { argument
76 struct GpuComplexT<std::complex<double>> {
85 inline const typename GpuComplexT<T>::type* GpuComplex(const T* p) {
86 return reinterpret_cast<const typename GpuComplexT<T>::type*>(p);
90 inline typename GpuComplexT<T>::type* GpuComplex(T* p) {
91 return reinterpret_cast<typename GpuComplexT<T>::type*>(p);
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2158 typedef typename HalfAsFloat<typename GpuComplexT<T>::type>::type CUDA_T; in DoBlasGemmBatchedInternal()