Searched refs:CudaScalar (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 354 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GeamImpl() typedef
356 reinterpret_cast<const CudaScalar*>(alpha), in GeamImpl()
357 reinterpret_cast<const CudaScalar*>(A), lda, in GeamImpl()
358 reinterpret_cast<const CudaScalar*>(beta), in GeamImpl()
359 reinterpret_cast<const CudaScalar*>(B), ldb, in GeamImpl()
360 reinterpret_cast<CudaScalar*>(C), ldc)); in GeamImpl()
419 using CudaScalar = typename CUDAComplexT<Scalar>::type; in PotrfBatchedImpl() typedef
421 cuda_solver->GetScratchSpace<uint8>(sizeof(CudaScalar*) * batch_size, "", in PotrfBatchedImpl()
429 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in PotrfBatchedImpl()
765 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetrfBatchedImpl() typedef
[all …]
|