| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | deterministic_test.py | 147 for sample_shape in [(), (4,)]:
149 sample = deterministic.sample(sample_shape)
150 self.assertAllEqual(sample_shape, sample.get_shape())
152 np.zeros(sample_shape).astype(np.float32), sample.eval())
156 for sample_shape in [(), (4,)]:
158 sample = deterministic.sample(sample_shape)
159 self.assertAllEqual(sample_shape + (2,), sample.get_shape())
161 np.zeros(sample_shape + (2,)).astype(np.float32), sample.eval())
165 sample_shape = array_ops.placeholder(np.int32)
170 sample_ = deterministic.sample(sample_shape).eval(
[all …]
|
| D | distribution_test.py | 50 for sample_shape in sample_shapes:
51 param_shapes = cls.param_shapes(sample_shape)
55 self.assertAllEqual(sample_shape,
58 self.assertAllEqual(sample_shape,
157 sample_shape = ops.convert_to_tensor([6, 7], dtype=dtypes.int32)
158 y = dist._set_sample_static_shape(x, sample_shape)
166 sample_shape = ops.convert_to_tensor([6, 7], dtype=dtypes.int32)
167 y = dist._set_sample_static_shape(x, sample_shape)
173 sample_shape = ops.convert_to_tensor([6, 7], dtype=dtypes.int32)
174 y = dist._set_sample_static_shape(x, sample_shape)
[all …]
|
| D | half_normal_test.py | 57 def _testParamShapes(self, sample_shape, expected): argument
59 param_shapes = hn_lib.HalfNormal.param_shapes(sample_shape)
67 def _testParamStaticShapes(self, sample_shape, expected): argument
68 param_shapes = hn_lib.HalfNormal.param_static_shapes(sample_shape)
79 sample_shape = [10, 3, 4]
80 self._testParamShapes(sample_shape, sample_shape)
81 self._testParamShapes(constant_op.constant(sample_shape), sample_shape)
84 sample_shape = [10, 3, 4]
85 self._testParamStaticShapes(sample_shape, sample_shape)
87 tensor_shape.TensorShape(sample_shape), sample_shape)
|
| D | cauchy_test.py | 58 def _testParamShapes(self, sample_shape, expected): argument
60 param_shapes = cauchy_lib.Cauchy.param_shapes(sample_shape)
70 def _testParamStaticShapes(self, sample_shape, expected): argument
71 param_shapes = cauchy_lib.Cauchy.param_static_shapes(sample_shape)
77 sample_shape = [10, 3, 4]
78 self._testParamShapes(sample_shape, sample_shape)
79 self._testParamShapes(constant_op.constant(sample_shape), sample_shape)
82 sample_shape = [10, 3, 4]
83 self._testParamStaticShapes(sample_shape, sample_shape)
85 tensor_shape.TensorShape(sample_shape), sample_shape)
|
| D | independent_test.py | 130 sample_shape = [4, 5]
144 x = ind.sample(sample_shape, seed=42)
170 self.assertAllEqual(sample_shape + batch_shape + image_shape, x_shape)
171 self.assertAllEqual(sample_shape + batch_shape, log_prob_x_shape)
|
| D | shape_test.py | 49 def _random_sample(self, sample_shape, dtype=np.float32): argument
50 return self._rng.random_sample(sample_shape).astype(dtype)
57 sample_shape = x_shape[:n]
64 return y, sample_shape
69 y, sample_shape = shaper.make_batch_of_event_sample_matrices(
72 y, sample_shape, expand_batch_dim=expand_batch_dim)
73 return y, sample_shape, should_be_x_value
524 def _random_sample(self, sample_shape, dtype=dtypes.float64): argument
525 return self._rng.random_sample(sample_shape).astype(dtype.as_numpy_dtype())
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | distribution.py | 425 def param_shapes(cls, sample_shape, name="DistributionParamShapes"): argument
442 with ops.name_scope(name, values=[sample_shape]):
443 return cls._param_shapes(sample_shape)
446 def param_static_shapes(cls, sample_shape): argument
467 if isinstance(sample_shape, tensor_shape.TensorShape):
468 if not sample_shape.is_fully_defined():
470 sample_shape = sample_shape.as_list()
472 params = cls.param_shapes(sample_shape)
485 def _param_shapes(sample_shape): argument
662 def _call_sample_n(self, sample_shape, seed, name, **kwargs): argument
[all …]
|
| D | transformed_distribution.py | 379 sample_shape = _concat_vectors(
384 x = self.distribution.sample(sample_shape=sample_shape, seed=seed)
389 def _call_sample_n(self, sample_shape, seed, name, **kwargs): argument
393 with self._name_scope(name, values=[sample_shape]):
394 sample_shape = ops.convert_to_tensor(
395 sample_shape, dtype=dtypes.int32, name="sample_shape")
396 sample_shape, n = self._expand_sample_shape_to_vector(
397 sample_shape, "sample_shape")
407 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
414 y = self._set_sample_static_shape(y, sample_shape)
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | conditional_transformed_distribution.py | 56 sample_shape = _concat_vectors(
62 x = self.distribution.sample(sample_shape=sample_shape,
69 def _call_sample_n(self, sample_shape, seed, name, argument
75 with self._name_scope(name, values=[sample_shape]):
76 sample_shape = ops.convert_to_tensor(
77 sample_shape, dtype=dtypes.int32, name="sample_shape")
78 sample_shape, n = self._expand_sample_shape_to_vector(
79 sample_shape, "sample_shape")
89 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
97 y = self._set_sample_static_shape(y, sample_shape)
|
| D | shape.py | 384 sample_shape, batch_shape, event_shape = self.get_shape(x)
395 return x, sample_shape
400 self, x, sample_shape, expand_batch_dim=True, argument
421 with self._name_scope(name, values=[x, sample_shape]):
424 sample_shape = ops.convert_to_tensor(sample_shape, name="sample_shape")
448 new_shape = array_ops.concat([sample_shape, batch_shape, event_shape], 0)
|
| D | conditional_distribution.py | 35 def sample(self, sample_shape=(), seed=None, name="sample", argument
37 return self._call_sample_n(sample_shape, seed, name, **condition_kwargs)
|
| D | chi2.py | 100 def _param_shapes(sample_shape): argument
101 return {"df": ops.convert_to_tensor(sample_shape, dtype=dtypes.int32)}
|
| D | relaxed_bernoulli.py | 188 def _param_shapes(sample_shape): argument
189 return {"logits": ops.convert_to_tensor(sample_shape, dtype=dtypes.int32)}
|
| /external/tensorflow/tensorflow/contrib/bayesflow/python/kernel_tests/ |
| D | mcmc_diagnostics_test.py | 301 sample_shape, argument
307 state_shape = sample_shape + independent_chain_shape + other_shape
319 sample_shape=[10000], independent_chain_shape=[4], other_shape=[])
323 sample_shape=[10000], independent_chain_shape=[3], other_shape=[7])
327 sample_shape=[10000], independent_chain_shape=[2], other_shape=[5, 7])
331 sample_shape=[10000],
337 self, sample_shape, independent_chain_shape, other_shape): argument
340 state_shape = sample_shape + independent_chain_shape + other_shape
346 sample_shape) + independent_chain_shape + [1] * len(other_shape))
353 sample_shape=[10000], independent_chain_shape=[2], other_shape=[5])
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/preprocessing/ |
| D | sequence.py | 75 sample_shape = tuple()
78 sample_shape = np.asarray(s).shape[1:]
81 x = (np.ones((num_samples, maxlen) + sample_shape) * value).astype(dtype)
94 if trunc.shape[1:] != sample_shape:
98 % (trunc.shape[1:], idx, sample_shape))
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | normal_test.py | 61 def _testParamShapes(self, sample_shape, expected): argument
63 param_shapes = normal_lib.Normal.param_shapes(sample_shape)
73 def _testParamStaticShapes(self, sample_shape, expected): argument
74 param_shapes = normal_lib.Normal.param_static_shapes(sample_shape)
80 sample_shape = [10, 3, 4]
81 self._testParamShapes(sample_shape, sample_shape)
82 self._testParamShapes(constant_op.constant(sample_shape), sample_shape)
85 sample_shape = [10, 3, 4]
86 self._testParamStaticShapes(sample_shape, sample_shape)
88 tensor_shape.TensorShape(sample_shape), sample_shape)
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | affine_linear_operator.py | 192 y, sample_shape = self._shaper.make_batch_of_event_sample_matrices(
198 y, sample_shape, expand_batch_dim=False)
208 x, sample_shape = self._shaper.make_batch_of_event_sample_matrices(
213 x, sample_shape, expand_batch_dim=False)
|
| D | affine.py | 358 y, sample_shape = self._shaper.make_batch_of_event_sample_matrices(
364 y, sample_shape, expand_batch_dim=False)
376 x, sample_shape = self._shaper.make_batch_of_event_sample_matrices(
381 x, sample_shape, expand_batch_dim=False)
|
| /external/tensorflow/tensorflow/tools/api/golden/ |
| D | tensorflow.distributions.-distribution.pbtxt | 104 …argspec: "args=[\'cls\', \'sample_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'Dis…
108 argspec: "args=[\'cls\', \'sample_shape\'], varargs=None, keywords=None, defaults=None"
120 …argspec: "args=[\'self\', \'sample_shape\', \'seed\', \'name\'], varargs=None, keywords=None, defa…
|
| D | tensorflow.distributions.-bernoulli.pbtxt | 113 …argspec: "args=[\'cls\', \'sample_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'Dis…
117 argspec: "args=[\'cls\', \'sample_shape\'], varargs=None, keywords=None, defaults=None"
129 …argspec: "args=[\'self\', \'sample_shape\', \'seed\', \'name\'], varargs=None, keywords=None, defa…
|
| D | tensorflow.distributions.-beta.pbtxt | 117 …argspec: "args=[\'cls\', \'sample_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'Dis…
121 argspec: "args=[\'cls\', \'sample_shape\'], varargs=None, keywords=None, defaults=None"
133 …argspec: "args=[\'self\', \'sample_shape\', \'seed\', \'name\'], varargs=None, keywords=None, defa…
|
| D | tensorflow.distributions.-exponential.pbtxt | 114 …argspec: "args=[\'cls\', \'sample_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'Dis…
118 argspec: "args=[\'cls\', \'sample_shape\'], varargs=None, keywords=None, defaults=None"
130 …argspec: "args=[\'self\', \'sample_shape\', \'seed\', \'name\'], varargs=None, keywords=None, defa…
|
| D | tensorflow.distributions.-gamma.pbtxt | 113 …argspec: "args=[\'cls\', \'sample_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'Dis…
117 argspec: "args=[\'cls\', \'sample_shape\'], varargs=None, keywords=None, defaults=None"
129 …argspec: "args=[\'self\', \'sample_shape\', \'seed\', \'name\'], varargs=None, keywords=None, defa…
|
| D | tensorflow.distributions.-multinomial.pbtxt | 117 …argspec: "args=[\'cls\', \'sample_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'Dis…
121 argspec: "args=[\'cls\', \'sample_shape\'], varargs=None, keywords=None, defaults=None"
133 …argspec: "args=[\'self\', \'sample_shape\', \'seed\', \'name\'], varargs=None, keywords=None, defa…
|
| D | tensorflow.distributions.-normal.pbtxt | 113 …argspec: "args=[\'cls\', \'sample_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'Dis…
117 argspec: "args=[\'cls\', \'sample_shape\'], varargs=None, keywords=None, defaults=None"
129 …argspec: "args=[\'self\', \'sample_shape\', \'seed\', \'name\'], varargs=None, keywords=None, defa…
|