MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GpuMemoryMutable

Function GpuMemoryMutable

tensorflow/stream_executor/gpu/gpu_helpers.h:47–49  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

45// pointer in CUDA device memory.
46template <typename T>
47T* GpuMemoryMutable(DeviceMemory<T>* mem) {
48 return static_cast<T*>(mem->opaque());
49}
50
51static_assert(
52 sizeof(std::complex<float>) == sizeof(GpuComplexType),

Callers 15

DoBlasAsumMethod · 0.85
DoBlasAxpyMethod · 0.85
DoBlasCopyMethod · 0.85
DoBlasDotMethod · 0.85
DoBlasNrm2Method · 0.85
DoBlasScalMethod · 0.85
DoBlasSwapMethod · 0.85
DoBlasIamaxMethod · 0.85
DoBlasIaminMethod · 0.85
DoBlasGemvMethod · 0.85
DoBlasGerMethod · 0.85
DoBlasSyrMethod · 0.85

Calls 1

opaqueMethod · 0.80

Tested by

no test coverage detected