OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:CUDAComplex
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/stream_executor/cuda/
cuda_blas.cc
662
CUDAComplex
(CUDAMemory(x)), incx, CUDAMemoryMutable(result));
670
CUDAComplex
(CUDAMemory(x)), incx, CUDAMemoryMutable(result));
695
CUDAComplex
(&alpha),
CUDAComplex
(CUDAMemory(x)), incx,
696
CUDAComplex
(CUDAMemoryMutable(y)), incy);
705
CUDAComplex
(&alpha),
CUDAComplex
(CUDAMemory(x)), incx,
706
CUDAComplex
(CUDAMemoryMutable(y)), incy);
730
CUDAComplex
(CUDAMemory(x)), incx,
731
CUDAComplex
(CUDAMemoryMutable(y)), incy)
[
all
...]
cuda_helpers.h
84
inline const typename CUDAComplexT<T>::type *
CUDAComplex
(const T *p) {
89
inline typename CUDAComplexT<T>::type *
CUDAComplex
(T *p) {
cuda_fft.cc
478
CUDAComplex
(const_cast<InputT *>(CUDAMemory(input))),
479
CUDAComplex
(CUDAMemoryMutable(output)));
505
CUDAComplex
(const_cast<InputT *>(CUDAMemory(input))),
506
CUDAComplex
(CUDAMemoryMutable(output)),
/external/tensorflow/tensorflow/core/kernels/
cuda_solvers.cc
368
bufsize(cusolver_dn_handle, uplo, n,
CUDAComplex
(A), lda, &lwork));
374
cusolver_dn_handle, uplo, n,
CUDAComplex
(A), lda,
375
CUDAComplex
(dev_workspace.mutable_data()), lwork, dev_lapack_info));
401
bufsize(cusolver_dn_handle, m, n,
CUDAComplex
(A), lda, &lwork));
407
cusolver_dn_handle, m, n,
CUDAComplex
(A), lda,
408
CUDAComplex
(dev_workspace.mutable_data()), dev_pivots, dev_lapack_info));
433
CUDAComplex
(A), lda, pivots,
434
CUDAComplex
(B), ldb, dev_lapack_info));
461
bufsize(cusolver_dn_handle, m, n,
CUDAComplex
(A), lda, &lwork));
467
cusolver_dn_handle, m, n,
CUDAComplex
(A), lda, CUDAComplex(tau)
[
all
...]
cuda_solvers.h
51
inline const typename CUDAComplexT<T>::type*
CUDAComplex
(const T* p) {
55
inline typename CUDAComplexT<T>::type*
CUDAComplex
(T* p) {
Completed in 133 milliseconds