Home
last modified time | relevance | path

Searched refs:GpuMemory (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc437 elem_count, GpuMemory(x), incx, in DoBlasAsum()
445 elem_count, GpuMemory(x), incx, in DoBlasAsum()
453 elem_count, GpuComplex(GpuMemory(x)), incx, in DoBlasAsum()
461 elem_count, GpuComplex(GpuMemory(x)), incx, in DoBlasAsum()
469 elem_count, &alpha, GpuMemory(x), incx, in DoBlasAxpy()
477 elem_count, &alpha, GpuMemory(x), incx, in DoBlasAxpy()
487 GpuComplex(GpuMemory(x)), incx, in DoBlasAxpy()
497 GpuComplex(GpuMemory(x)), incx, in DoBlasAxpy()
505 elem_count, GpuMemory(x), incx, GpuMemoryMutable(y), in DoBlasCopy()
513 elem_count, GpuMemory(x), incx, GpuMemoryMutable(y), in DoBlasCopy()
[all …]
Dcuda_fft.cc470 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftInternal()
498 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc405 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasAsum()
413 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasAsum()
437 GpuMemory(x), incx, GpuMemoryMutable(y), incy); in DoBlasAxpy()
445 GpuMemory(x), incx, GpuMemoryMutable(y), incy); in DoBlasAxpy()
471 GpuMemory(x), incx, GpuMemoryMutable(y), incy); in DoBlasCopy()
479 GpuMemory(x), incx, GpuMemoryMutable(y), incy); in DoBlasCopy()
504 GpuMemory(x), incx, GpuMemory(y), incy, GpuMemoryMutable(result)); in DoBlasDot()
513 GpuMemory(x), incx, GpuMemory(y), incy, GpuMemoryMutable(result)); in DoBlasDot()
557 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasNrm2()
565 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasNrm2()
[all …]
Drocm_fft.cc518 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftInternal()
545 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_helpers.h40 const T* GpuMemory(const DeviceMemory<T>& mem) { in GpuMemory() function