diff options
| author |  Manjunath Kudlur <keveman@gmail.com> | 2016-01-27 16:14:12 -0800 |
|---|---|---|
| committer |  Vijay Vasudevan <vrv@google.com> | 2016-01-27 23:06:51 -0800 |
| commit | 33aa62872774539f2a8144db0c93a1b9ce3ed51f (patch) | |
| tree | 81dac4e42b01bae2b9ddc7e2eb188ad6a8f882e0 /tensorflow/core/graph/testlib.cc | |
| parent | 6aac6d70793a7a7ce8e7c71c87e5cb610cf0f9af (diff) | |
Fixed constant folding to handle nodes with multiple outputs.
Change: 113212117
Diffstat (limited to 'tensorflow/core/graph/testlib.cc')
| -rw-r--r-- | tensorflow/core/graph/testlib.cc | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/core/graph/testlib.cc b/tensorflow/core/graph/testlib.cc index ab2e462e87..ae055803ee 100644 --- a/tensorflow/core/graph/testlib.cc +++ b/tensorflow/core/graph/testlib.cc @@ -308,6 +308,15 @@ Node* Cast(Graph* g, Node* in, DataType dst) { return ret; } +Node* BroadcastGradientArgs(Graph* g, Node* s0, Node* s1) { + Node* ret; + TF_CHECK_OK(NodeBuilder(g->NewName("n"), "BroadcastGradientArgs") + .Input(s0) + .Input(s1) + .Finalize(g, &ret)); + return ret; +} + void ToGraphDef(Graph* g, GraphDef* gdef) { g->ToGraphDef(gdef); } } // end namespace graph |