Searched refs:CUDABlas (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 234 bool CUDABlas::Init() { in Init()
245 CUDABlas::CUDABlas(gpu::GpuExecutor *parent) in CUDABlas() function in stream_executor::gpu::CUDABlas
248 CUDABlas::~CUDABlas() { in ~CUDABlas()
255 bool CUDABlas::SetStream(Stream *stream) { in SetStream()
402 bool CUDABlas::DoBlasInternalImpl(FuncT cublas_func, Stream *stream, in DoBlasInternalImpl()
433 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
441 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
449 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
457 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
465 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, float alpha, in DoBlasAxpy()
[all …]
|
| D | cuda_blas.h | 53 class CUDABlas : public blas::BlasSupport {
55 explicit CUDABlas(GpuExecutor *parent);
61 ~CUDABlas() override;
155 SE_DISALLOW_COPY_AND_ASSIGN(CUDABlas);
|