MCPcopy Create free account
hub / github.com/BVLC/caffe / caffe_gpu_memset

Function caffe_gpu_memset

include/caffe/util/math_functions.hpp:177–183  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

175void caffe_gpu_set(const int N, const Dtype alpha, Dtype *X);
176
177inline void caffe_gpu_memset(const size_t N, const int alpha, void* X) {
178#ifndef CPU_ONLY
179 CUDA_CHECK(cudaMemset(X, alpha, N)); // NOLINT(caffe/alt_fn)
180#else
181 NO_GPU;
182#endif
183}
184
185template <typename Dtype>
186void caffe_gpu_add_scalar(const int N, const Dtype alpha, Dtype *X);

Callers 2

to_gpuMethod · 0.85
TEST_FFunction · 0.85

Calls

no outgoing calls

Tested by 1

TEST_FFunction · 0.68