Searched refs:ROCMBlas (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 296 bool ROCMBlas::Init() { in Init()
306 ROCMBlas::ROCMBlas(gpu::GpuExecutor *parent) in ROCMBlas() function in stream_executor::gpu::ROCMBlas
309 ROCMBlas::~ROCMBlas() { in ~ROCMBlas()
315 bool ROCMBlas::SetStream(Stream *stream) { in SetStream()
382 bool ROCMBlas::DoBlasInternalImpl(FuncT rocblas_func, Stream *stream, in DoBlasInternalImpl()
400 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
408 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
416 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
424 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
432 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, float alpha, in DoBlasAxpy()
[all …]
|
| D | rocm_blas.h | 50 class ROCMBlas : public blas::BlasSupport {
52 explicit ROCMBlas(GpuExecutor *parent);
58 ~ROCMBlas() override;
153 SE_DISALLOW_COPY_AND_ASSIGN(ROCMBlas);
|