Support None trainable variables that don't produce a gradient in replicate_model_fn.
authorIgor Saprykin <isaprykin@google.com>
Tue, 13 Feb 2018 00:24:45 +0000 (16:24 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 13 Feb 2018 00:28:14 +0000 (16:28 -0800)
commit96564330fb0508a50a0515be11c9202c64b0f5b7
tree674d7a0cedeb48c1957786ffd9245072ecbc101d
parent4a1bcf7c4193906a4e77c1726657429e37579a65
Support None trainable variables that don't produce a gradient in replicate_model_fn.

This fixes #16829.

PiperOrigin-RevId: 185453911
tensorflow/contrib/estimator/python/estimator/replicate_model_fn.py
tensorflow/contrib/estimator/python/estimator/replicate_model_fn_test.py