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.cc581 elem_count, GpuMemory(x), incx, in DoBlasInternalImpl()
589 elem_count, GpuMemory(x), incx, in DoBlasInternalImpl()
597 elem_count, GpuComplex(GpuMemory(x)), incx, in DoBlasInternalImpl()
605 elem_count, GpuComplex(GpuMemory(x)), incx, in DoBlasInternalImpl()
613 elem_count, &alpha, GpuMemory(x), incx, in DoBlasInternalImpl()
621 elem_count, &alpha, GpuMemory(x), incx, in DoBlasInternalImpl()
632 GpuComplex(GpuMemory(x)), incx, in DoBlasInternalImpl()
643 GpuComplex(GpuMemory(x)), incx, in DoBlasInternalImpl()
651 elem_count, GpuMemory(x), incx, GpuMemoryMutable(y), in DoBlasInternalImpl()
659 elem_count, GpuMemory(x), incx, GpuMemoryMutable(y), in DoBlasInternalImpl()
[all …]
Dcuda_fft.cc499 GpuComplex(const_cast<InputT *>(GpuMemory(input_maybe_copy))), in DoFftInternal()
527 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc301 GpuMemory(a)); in complex_cast()
449 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasAsum()
457 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasAsum()
482 GpuMemory(x), incx, GpuMemoryMutable(y), incy); in DoBlasAxpy()
491 GpuMemory(x), incx, GpuMemoryMutable(y), incy); in DoBlasAxpy()
517 GpuMemory(x), incx, GpuMemoryMutable(y), incy); in DoBlasCopy()
525 GpuMemory(x), incx, GpuMemoryMutable(y), incy); in DoBlasCopy()
551 GpuMemory(x), incx, GpuMemory(y), incy, GpuMemoryMutable(result)); in DoBlasDot()
561 GpuMemory(x), incx, GpuMemory(y), incy, GpuMemoryMutable(result)); in DoBlasDot()
605 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasNrm2()
[all …]
Drocm_fft.cc523 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftInternal()
550 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_helpers.h42 const T* GpuMemory(const DeviceMemory<T>& mem) { in GpuMemory() function