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.h | |
| parent | 6aac6d70793a7a7ce8e7c71c87e5cb610cf0f9af (diff) | |
Fixed constant folding to handle nodes with multiple outputs.
Change: 113212117
Diffstat (limited to 'tensorflow/core/graph/testlib.h')
| -rw-r--r-- | tensorflow/core/graph/testlib.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/graph/testlib.h b/tensorflow/core/graph/testlib.h index 8e98b1e57a..6bb209a1ea 100644 --- a/tensorflow/core/graph/testlib.h +++ b/tensorflow/core/graph/testlib.h @@ -147,6 +147,9 @@ Node* Select(Graph* g, Node* c, Node* inx, Node* iny); // Casts "in" into data type "dst". Node* Cast(Graph* g, Node* in, DataType dst); +// Computes the args needed broadcast gradient function. +Node* BroadcastGradientArgs(Graph* g, Node* s0, Node* s1); + } // end namespace graph } // end namespace test } // end namespace tensorflow |