Name tensors created in DistributionWith* subclasses.
Change: 144112386
This commit is contained in:
parent
28fd303882
commit
b7fc7f3f46
@ -183,7 +183,7 @@ class BernoulliWithSigmoidP(Bernoulli):
|
||||
parameters.pop("self")
|
||||
with ops.name_scope(name) as ns:
|
||||
super(BernoulliWithSigmoidP, self).__init__(
|
||||
p=nn.sigmoid(p),
|
||||
p=nn.sigmoid(p, name="sigmoid_p"),
|
||||
dtype=dtype,
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
|
@ -294,8 +294,8 @@ class BetaWithSoftplusAB(Beta):
|
||||
parameters.pop("self")
|
||||
with ops.name_scope(name, values=[a, b]) as ns:
|
||||
super(BetaWithSoftplusAB, self).__init__(
|
||||
a=nn.softplus(a),
|
||||
b=nn.softplus(b),
|
||||
a=nn.softplus(a, name="softplus_a"),
|
||||
b=nn.softplus(b, name="softplus_b"),
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
name=ns)
|
||||
|
@ -93,7 +93,8 @@ class Chi2WithAbsDf(Chi2):
|
||||
parameters.pop("self")
|
||||
with ops.name_scope(name, values=[df]) as ns:
|
||||
super(Chi2WithAbsDf, self).__init__(
|
||||
df=math_ops.floor(math_ops.abs(df)),
|
||||
df=math_ops.floor(math_ops.abs(df, name="abs_df"),
|
||||
name="floor_abs_df"),
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
name=ns)
|
||||
|
@ -113,7 +113,7 @@ class ExponentialWithSoftplusLam(Exponential):
|
||||
parameters.pop("self")
|
||||
with ops.name_scope(name, values=[lam]) as ns:
|
||||
super(ExponentialWithSoftplusLam, self).__init__(
|
||||
lam=nn.softplus(lam),
|
||||
lam=nn.softplus(lam, name="softplus_lam"),
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
name=ns)
|
||||
|
@ -234,8 +234,8 @@ class GammaWithSoftplusAlphaBeta(Gamma):
|
||||
parameters.pop("self")
|
||||
with ops.name_scope(name, values=[alpha, beta]) as ns:
|
||||
super(GammaWithSoftplusAlphaBeta, self).__init__(
|
||||
alpha=nn.softplus(alpha),
|
||||
beta=nn.softplus(beta),
|
||||
alpha=nn.softplus(alpha, name="softplus_alpha"),
|
||||
beta=nn.softplus(beta, name="softplus_beta"),
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
name=ns)
|
||||
|
@ -232,8 +232,8 @@ class InverseGammaWithSoftplusAlphaBeta(InverseGamma):
|
||||
parameters.pop("self")
|
||||
with ops.name_scope(name, values=[alpha, beta]) as ns:
|
||||
super(InverseGammaWithSoftplusAlphaBeta, self).__init__(
|
||||
alpha=nn.softplus(alpha),
|
||||
beta=nn.softplus(beta),
|
||||
alpha=nn.softplus(alpha, name="softplus_alpha"),
|
||||
beta=nn.softplus(beta, name="softplus_gamma"),
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
name=ns)
|
||||
|
@ -188,7 +188,7 @@ class LaplaceWithSoftplusScale(Laplace):
|
||||
with ops.name_scope(name, values=[loc, scale]) as ns:
|
||||
super(LaplaceWithSoftplusScale, self).__init__(
|
||||
loc=loc,
|
||||
scale=nn.softplus(scale),
|
||||
scale=nn.softplus(scale, name="softplus_scale"),
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
name=ns)
|
||||
|
@ -218,7 +218,7 @@ class NormalWithSoftplusSigma(Normal):
|
||||
with ops.name_scope(name, values=[sigma]) as ns:
|
||||
super(NormalWithSoftplusSigma, self).__init__(
|
||||
mu=mu,
|
||||
sigma=nn.softplus(sigma),
|
||||
sigma=nn.softplus(sigma, name="softplus_sigma"),
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
name=ns)
|
||||
|
@ -333,7 +333,7 @@ class StudentTWithAbsDfSoftplusSigma(StudentT):
|
||||
super(StudentTWithAbsDfSoftplusSigma, self).__init__(
|
||||
df=math_ops.floor(math_ops.abs(df)),
|
||||
mu=mu,
|
||||
sigma=nn.softplus(sigma),
|
||||
sigma=nn.softplus(sigma, name="softplus_sigma"),
|
||||
validate_args=validate_args,
|
||||
allow_nan_stats=allow_nan_stats,
|
||||
name=ns)
|
||||
|
Loading…
Reference in New Issue
Block a user