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.h68 struct GpuComplexT { struct
73 struct GpuComplexT<std::complex<float>> { argument
78 struct GpuComplexT<std::complex<double>> {
87 inline const typename GpuComplexT<T>::type* GpuComplex(const T* p) {
88 auto* result = reinterpret_cast<const typename GpuComplexT<T>::type*>(p);
95 inline typename GpuComplexT<T>::type* GpuComplex(T* p) {
96 auto* result = reinterpret_cast<typename GpuComplexT<T>::type*>(p);
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2378 typedef typename HalfAsFloat<typename GpuComplexT<T>::type>::type CUDA_T; in DoBlasInternalImpl()