zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

minor fixes (#1194)

Browse Source 
* minor fixes

* fix build errors
This commit is contained in:
Fangjun Kuang
2024-06-13 13:06:49 +08:00
committed by GitHub
parent 934683088e
commit f20e97b092
16 changed files with 239 additions and 238 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mlx/random.h
View File

@@ -148,7 +148,7 @@ array randint(
const std::optional<array>& key = std::nullopt,
StreamOrDevice s = {}) {
return randint(array(low), array(high), shape, dtype, key, to_stream(s));
};
}
/** Generate binary variables with probability to be true equal to p */
array bernoulli(
@@ -167,7 +167,7 @@ array bernoulli(
const std::optional<array>& key = std::nullopt,
StreamOrDevice s = {}) {
return bernoulli(array(p), key, s);
};
}
template <typename T>
array bernoulli(
@@ -176,7 +176,7 @@ array bernoulli(
const std::optional<array>& key = std::nullopt,
StreamOrDevice s = {}) {
return bernoulli(array(p), shape, key, s);
};
}
array bernoulli(
const std::optional<array>& key = std::nullopt,