Searched refs:perturb_factor (Results 1 – 1 of 1) sorted by relevance
227 perturb_factor=scale_perturb_factor,266 perturb_diag, perturb_factor, shift, argument297 perturb_factor = _as_tensor(perturb_factor, "perturb_factor")303 if perturb_factor is not None:304 shape_hint = distribution_util.dimension_size(perturb_factor, axis=-2)325 if perturb_factor is not None:328 u=perturb_factor,