Searched refs:check_distribution_name (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/nn/probability/distribution/ |
| D | log_normal.py | 21 from ._utils.utils import check_distribution_name
218 check_distribution_name(dist, 'LogNormal')
236 check_distribution_name(dist, 'LogNormal')
|
| D | gumbel.py | 24 from ._utils.utils import check_distribution_name
200 check_distribution_name(dist, 'Gumbel')
216 check_distribution_name(dist, 'Gumbel')
|
| D | normal.py | 22 from ._utils.utils import check_greater_zero, check_distribution_name
230 check_distribution_name(dist, 'Normal')
288 check_distribution_name(dist, 'Normal')
|
| D | exponential.py | 22 from ._utils.utils import check_greater_zero, check_distribution_name
224 check_distribution_name(dist, 'Exponential')
304 check_distribution_name(dist, 'Exponential')
|
| D | geometric.py | 22 from ._utils.utils import check_prob, check_distribution_name
234 check_distribution_name(dist, 'Geometric')
293 check_distribution_name(dist, 'Geometric')
|
| D | bernoulli.py | 21 from ._utils.utils import check_prob, check_distribution_name, clamp_probs
227 check_distribution_name(dist, 'Bernoulli')
293 check_distribution_name(dist, 'Bernoulli')
|
| D | gamma.py | 23 from ._utils.utils import check_greater_zero, check_distribution_name
273 check_distribution_name(dist, 'Gamma')
330 check_distribution_name(dist, 'Gamma')
|
| D | cauchy.py | 22 from ._utils.utils import check_greater_zero, check_distribution_name, raise_not_defined
314 check_distribution_name(dist, 'Cauchy')
336 check_distribution_name(dist, 'Cauchy')
|
| D | beta.py | 23 from ._utils.utils import check_greater_zero, check_distribution_name
275 check_distribution_name(dist, 'Beta')
315 check_distribution_name(dist, 'Beta')
|
| D | uniform.py | 22 from ._utils.utils import check_greater, check_distribution_name
253 check_distribution_name(dist, 'Uniform')
293 check_distribution_name(dist, 'Uniform')
|
| D | categorical.py | 25 check_distribution_name, raise_not_implemented_util
238 check_distribution_name(dist, 'Categorical')
256 check_distribution_name(dist, 'Categorical')
|
| /third_party/mindspore/mindspore/nn/probability/distribution/_utils/ |
| D | utils.py | 271 def check_distribution_name(name, expected_name): function
|