| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/distributions/ |
| H A D | distribution.py | 491 def param_shapes(cls, sample_shape, name="DistributionParamShapes"): argument
512 def param_static_shapes(cls, sample_shape): argument
551 def _param_shapes(sample_shape): argument
732 def _call_sample_n(self, sample_shape, seed, name, **kwargs): argument
745 def sample(self, sample_shape=(), seed=None, name="sample"): argument
1261 def _set_sample_static_shape(self, x, sample_shape): argument
|
| H A D | exponential.py | 115 def _param_shapes(sample_shape): argument
|
| H A D | uniform.py | 130 def _param_shapes(sample_shape): argument
|
| H A D | bernoulli.py | 96 def _param_shapes(sample_shape): argument
|
| H A D | student_t.py | 206 def _param_shapes(sample_shape): argument
|
| H A D | gamma.py | 183 def _param_shapes(sample_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| H A D | normal_test.py | 59 def _testParamShapes(self, sample_shape, expected): argument
70 def _testParamStaticShapes(self, sample_shape, expected): argument
|
| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | lkj_cholesky.py | 97 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | poisson.py | 63 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | geometric.py | 106 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | transformed_distribution.py | 133 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | multinomial.py | 101 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | negative_binomial.py | 110 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | mixture_same_family.py | 171 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | independent.py | 103 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | categorical.py | 130 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | binomial.py | 118 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | bernoulli.py | 104 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | normal.py | 70 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | von_mises.py | 164 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | one_hot_categorical.py | 94 def sample(self, sample_shape=torch.Size()): argument
|
| H A D | wishart.py | 222 def _bartlett_sampling(self, sample_shape=torch.Size()): argument
|
| H A D | continuous_bernoulli.py | 165 def sample(self, sample_shape=torch.Size()): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/ |
| H A D | random_binomial_test.py | 36 self, num, counts, probs, dtype, gen=None, sample_shape=None, seed=None): argument
|
| /aosp_15_r20/external/pytorch/test/distributions/ |
| H A D | test_transforms.py | 506 base_batch_dim, base_event_dim, transform_dim, num_transforms, sample_shape argument
|