Home
last modified time | relevance | path

Searched refs:GpuMemoryMutable (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc582 GpuMemoryMutable(result)); in DoBlasInternalImpl()
590 GpuMemoryMutable(result)); in DoBlasInternalImpl()
598 GpuMemoryMutable(result)); in DoBlasInternalImpl()
606 GpuMemoryMutable(result)); in DoBlasInternalImpl()
614 GpuMemoryMutable(y), incy); in DoBlasInternalImpl()
622 GpuMemoryMutable(y), incy); in DoBlasInternalImpl()
633 GpuComplex(GpuMemoryMutable(y)), incy); in DoBlasInternalImpl()
644 GpuComplex(GpuMemoryMutable(y)), incy); in DoBlasInternalImpl()
651 elem_count, GpuMemory(x), incx, GpuMemoryMutable(y), in DoBlasInternalImpl()
659 elem_count, GpuMemory(x), incx, GpuMemoryMutable(y), in DoBlasInternalImpl()
[all …]
Dcuda_rng.cc129 rng_, reinterpret_cast<float*>(GpuMemoryMutable(v)), element_count); in DoPopulateRandUniformInternal()
132 rng_, reinterpret_cast<double*>(GpuMemoryMutable(v)), element_count); in DoPopulateRandUniformInternal()
176 func(rng_, GpuMemoryMutable(v), element_count, mean, stddev); in DoPopulateRandGaussianInternal()
Dcuda_fft.cc500 GpuComplex(GpuMemoryMutable(output))); in DoFftInternal()
528 GpuComplex(GpuMemoryMutable(output)), in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc314 GpuMemoryMutable(a)); in complex_cast()
449 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasAsum()
457 GpuMemory(x), incx, GpuMemoryMutable(result)); in DoBlasAsum()
465 complex_cast(x), incx, GpuMemoryMutable(result)); in DoBlasAsum()
473 complex_cast(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()
[all …]
Drocm_rng.cc182 parent_, rng_, reinterpret_cast<float*>(GpuMemoryMutable(v)), in DoPopulateRandUniformInternal()
186 parent_, rng_, reinterpret_cast<double*>(GpuMemoryMutable(v)), in DoPopulateRandUniformInternal()
230 func(parent_, rng_, GpuMemoryMutable(v), element_count, mean, stddev); in DoPopulateRandGaussianInternal()
Drocm_fft.cc524 GpuComplex(GpuMemoryMutable(output))); in DoFftInternal()
551 GpuComplex(GpuMemoryMutable(output)), in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_helpers.h49 T* GpuMemoryMutable(DeviceMemory<T>* mem) { in GpuMemoryMutable() function