HomeSort by relevance Sort by last modified time
    Searched refs:CUDABlas (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_blas.cc 234 bool CUDABlas::Init() {
245 CUDABlas::CUDABlas(gpu::GpuExecutor *parent)
248 CUDABlas::~CUDABlas() {
255 bool CUDABlas::SetStream(Stream *stream) {
402 bool CUDABlas::DoBlasInternalImpl(FuncT cublas_func, Stream *stream,
433 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count,
441 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count,
449 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count
    [all...]
cuda_blas.h 53 class CUDABlas : public blas::BlasSupport {
55 explicit CUDABlas(GpuExecutor *parent);
61 ~CUDABlas() override;
148 // GpuExecutor which instantiated this CUDABlas.
155 SE_DISALLOW_COPY_AND_ASSIGN(CUDABlas);

Completed in 305 milliseconds