Home
last modified time | relevance | path

Searched refs:CUDAComplexT (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_context.cc27 struct CUDAComplexT { struct
31 struct CUDAComplexT<std::complex<float>> { struct
35 struct CUDAComplexT<std::complex<double>> { struct
40 inline typename CUDAComplexT<T>::type* ToDevicePointer(se::DeviceMemory<T> p) { in ToDevicePointer()
41 return static_cast<typename CUDAComplexT<T>::type*>(p.opaque()); in ToDevicePointer()
/external/tensorflow/tensorflow/core/kernels/
Dcuda_solvers.h40 struct CUDAComplexT { struct
44 struct CUDAComplexT<std::complex<float>> { argument
48 struct CUDAComplexT<std::complex<double>> {
54 inline const typename CUDAComplexT<T>::type* CUDAComplex(const T* p) {
55 return reinterpret_cast<const typename CUDAComplexT<T>::type*>(p);
58 inline typename CUDAComplexT<T>::type* CUDAComplex(T* p) {
59 return reinterpret_cast<typename CUDAComplexT<T>::type*>(p);
Dcuda_solvers.cc333 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GeamImpl()
707 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetrfBatchedImpl()
742 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetrsBatchedImpl()
788 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetriBatchedImpl()
831 using CudaScalar = typename CUDAComplexT<Scalar>::type; in MatInvBatchedImpl()