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/util/
Dcuda_solvers.h45 struct CUDAComplexT { struct
49 struct CUDAComplexT<std::complex<float>> { argument
53 struct CUDAComplexT<std::complex<double>> {
59 inline const typename CUDAComplexT<T>::type* CUDAComplex(const T* p) {
60 return reinterpret_cast<const typename CUDAComplexT<T>::type*>(p);
63 inline typename CUDAComplexT<T>::type* CUDAComplex(T* p) {
64 return reinterpret_cast<typename CUDAComplexT<T>::type*>(p);
Dcuda_solvers.cc354 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GeamImpl()
419 using CudaScalar = typename CUDAComplexT<Scalar>::type; in PotrfBatchedImpl()
765 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetrfBatchedImpl()
800 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetrsBatchedImpl()
846 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetriBatchedImpl()
889 using CudaScalar = typename CUDAComplexT<Scalar>::type; in MatInvBatchedImpl()
932 using CudaScalar = typename CUDAComplexT<Scalar>::type; in TrsmImpl()
959 using CudaScalar = typename CUDAComplexT<Scalar>::type; in TrsvImpl()
985 using CudaScalar = typename CUDAComplexT<Scalar>::type; in TrsmBatchedImpl()