HomeSort by relevance Sort by last modified time
    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