Home
last modified time | relevance | path

Searched defs:sample_shape (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/distributions/
Ddistribution.py491 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
Dexponential.py115 def _param_shapes(sample_shape): argument
Duniform.py130 def _param_shapes(sample_shape): argument
Dbernoulli.py96 def _param_shapes(sample_shape): argument
Dstudent_t.py206 def _param_shapes(sample_shape): argument
Dgamma.py183 def _param_shapes(sample_shape): argument
/external/pytorch/torch/distributions/
Dlkj_cholesky.py97 def sample(self, sample_shape=torch.Size()): argument
Dpoisson.py63 def sample(self, sample_shape=torch.Size()): argument
Dgeometric.py106 def sample(self, sample_shape=torch.Size()): argument
Dtransformed_distribution.py133 def sample(self, sample_shape=torch.Size()): argument
Dmultinomial.py101 def sample(self, sample_shape=torch.Size()): argument
Dnegative_binomial.py110 def sample(self, sample_shape=torch.Size()): argument
Dindependent.py103 def sample(self, sample_shape=torch.Size()): argument
Dmixture_same_family.py171 def sample(self, sample_shape=torch.Size()): argument
Dcategorical.py130 def sample(self, sample_shape=torch.Size()): argument
Dbinomial.py118 def sample(self, sample_shape=torch.Size()): argument
Dbernoulli.py104 def sample(self, sample_shape=torch.Size()): argument
Dnormal.py70 def sample(self, sample_shape=torch.Size()): argument
Dvon_mises.py164 def sample(self, sample_shape=torch.Size()): argument
Done_hot_categorical.py94 def sample(self, sample_shape=torch.Size()): argument
Dwishart.py222 def _bartlett_sampling(self, sample_shape=torch.Size()): argument
Dcontinuous_bernoulli.py165 def sample(self, sample_shape=torch.Size()): argument
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dnormal_test.py59 def _testParamShapes(self, sample_shape, expected): argument
70 def _testParamStaticShapes(self, sample_shape, expected): argument
/external/tensorflow/tensorflow/python/kernel_tests/random/
Drandom_binomial_test.py36 self, num, counts, probs, dtype, gen=None, sample_shape=None, seed=None): argument
/external/pytorch/test/distributions/
Dtest_transforms.py506 base_batch_dim, base_event_dim, transform_dim, num_transforms, sample_shape argument

12