Name tensors created in DistributionWith* subclasses.

Change: 144112386
This commit is contained in:
Joshua V. Dillon 2017-01-10 12:13:54 -08:00 committed by TensorFlower Gardener
parent 28fd303882
commit b7fc7f3f46
9 changed files with 13 additions and 12 deletions

View File

@ -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,

View File

@ -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)

View File

@ -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)

View File

@ -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)

View File

@ -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)

View File

@ -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)

View File

@ -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)

View File

@ -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)

View File

@ -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)