Enable DCHECK in shape inference to check that the shape function

has called set_output for each output tensor.
Change: 153472242
This commit is contained in:
A. Unique TensorFlower 2017-04-18 08:41:02 -08:00 committed by TensorFlower Gardener
parent a498546ddf
commit d8b65d4035

View File

@ -182,15 +182,12 @@ class InferenceContext {
if (!s.ok()) {
return AttachContext(s);
}
#if 0
// TODO(cwhipkey): enable this check
#ifndef NDEBUG
for (int i = 0; i < num_outputs(); ++i) {
DCHECK(output(i).IsSet()) << i << " for " << node_def().name()
<< " of type " << node_def().op();
}
#endif // NDEBUG
#endif
return s;
}