MCPcopy Create free account
hub / github.com/LBANN/lbann / gaussian_fill

Function gaussian_fill

src/utils/random.cpp:270–281  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

268
269template <typename TensorDataType>
270void gaussian_fill(El::AbstractDistMatrix<TensorDataType>& mat,
271 El::Int m,
272 El::Int n,
273 TensorDataType mean,
274 TensorDataType stddev)
275{
276#ifdef LBANN_DETERMINISTIC
277 gaussian_fill_procdet(mat, m, n, mean, stddev);
278#else
279 gaussian_fill_parallel(mat, m, n, mean, stddev);
280#endif // LBANN_DETERMINISTIC
281}
282
283template <typename TensorDataType>
284void bernoulli_fill(El::AbstractDistMatrix<TensorDataType>& mat,

Callers 4

fp_computeMethod · 0.85
fillMethod · 0.85
fillMethod · 0.85

Calls 2

gaussian_fill_procdetFunction · 0.85
gaussian_fill_parallelFunction · 0.85

Tested by

no test coverage detected