Searched refs:CudaScalar (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 333 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GeamImpl() typedef
335 reinterpret_cast<const CudaScalar*>(alpha), in GeamImpl()
336 reinterpret_cast<const CudaScalar*>(A), lda, in GeamImpl()
337 reinterpret_cast<const CudaScalar*>(beta), in GeamImpl()
338 reinterpret_cast<const CudaScalar*>(B), ldb, in GeamImpl()
339 reinterpret_cast<CudaScalar*>(C), ldc)); in GeamImpl()
707 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetrfBatchedImpl() typedef
709 cuda_solver->GetScratchSpace<uint8>(sizeof(CudaScalar*) * batch_size, "", in GetrfBatchedImpl()
717 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in GetrfBatchedImpl()
742 using CudaScalar = typename CUDAComplexT<Scalar>::type; in GetrsBatchedImpl() typedef
[all …]
|