Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcusolver_context.cc32 struct GpuComplexT { struct
33 typedef T type;
47 struct GpuComplexT<std::complex<float>> { struct
48 typedef cuComplex type;
51 struct GpuComplexT<std::complex<double>> { struct
52 typedef cuDoubleComplex type;
56 struct GpuComplexT<std::complex<float>*> { struct
57 typedef cuComplex* type;
60 struct GpuComplexT<std::complex<double>*> { struct
61 typedef cuDoubleComplex* type;
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_helpers.h67 struct GpuComplexT { struct
72 struct GpuComplexT<std::complex<float>> { argument