Searched refs:shape_hint (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | distribution_util_test.py | 63 shape_hint=None): argument
74 loc, scale_diag, scale_identity_multiplier, shape_hint)
81 shape_hint=None): argument
88 if loc is None and shape_hint is None:
91 if shape_hint is None:
92 shape_hint = loc.shape[-1]
95 return scale_identity_multiplier * np.diag(np.ones(shape_hint))
101 self, loc=None, shape_hint=None, scale_params=None): argument
108 "shape_hint": shape_hint}, **args)
122 shape_hint=2,
[all …]
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | distribution_util.py | 95 shape_hint=None, argument
180 shape_hint=shape_hint,
190 shape_hint=None, argument
255 if loc is None and shape_hint is None:
260 if shape_hint is None:
261 shape_hint = loc.shape[-1]
265 num_rows=shape_hint,
272 num_rows=shape_hint,
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | affine.py | 302 shape_hint = None
304 shape_hint = distribution_util.dimension_size(perturb_factor, axis=-2)
323 shape_hint=shape_hint)
|