| 268 | |
| 269 | template <typename TensorDataType> |
| 270 | void 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 | |
| 283 | template <typename TensorDataType> |
| 284 | void bernoulli_fill(El::AbstractDistMatrix<TensorDataType>& mat, |
no test coverage detected