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

Method DoBlasNrm2

tensorflow/stream_executor/cuda/cuda_blas.cc:749–755  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

747}
748
749bool CUDABlas::DoBlasNrm2(Stream *stream, uint64 elem_count,
750 const DeviceMemory<float> &x, int incx,
751 DeviceMemory<float> *result) {
752 return DoBlasInternal<float>(cublasSnrm2, stream,
753 false /* = pointer_mode_host */, elem_count,
754 GpuMemory(x), incx, GpuMemoryMutable(result));
755}
756
757bool CUDABlas::DoBlasNrm2(Stream *stream, uint64 elem_count,
758 const DeviceMemory<double> &x, int incx,

Callers

nothing calls this directly

Calls 3

GpuMemoryFunction · 0.85
GpuMemoryMutableFunction · 0.85
GpuComplexFunction · 0.85

Tested by

no test coverage detected