diff options
author | 2017-01-11 10:50:26 -0800 | |
---|---|---|
committer | 2017-01-11 11:06:43 -0800 | |
commit | 51e6e84fadd7b37c6e8e5c13cd0de4c7c8e2959f (patch) | |
tree | 1be0c95c807d02d34618a671864284b360556756 /tensorflow/core/graph/testlib.cc | |
parent | c14c1686c64bb93b4d372547f64680ff37433508 (diff) |
Add optimized reverse kernel for reversing rows of a 3D tensor. This is used in
image preprocessing pipelines.
Add benchmark for reverse.
Change: 144224184
Diffstat (limited to 'tensorflow/core/graph/testlib.cc')
-rw-r--r-- | tensorflow/core/graph/testlib.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/graph/testlib.cc b/tensorflow/core/graph/testlib.cc index 13c6a2146b..ef4dd04787 100644 --- a/tensorflow/core/graph/testlib.cc +++ b/tensorflow/core/graph/testlib.cc @@ -254,6 +254,10 @@ Node* Identity(Graph* g, Node* input, int index) { Node* Add(Graph* g, Node* in0, Node* in1) { return Binary(g, "Add", in0, in1); } +Node* Reverse(Graph* g, Node* tensor, Node* axis) { + return Binary(g, "ReverseV2", tensor, axis); +} + Node* Error(Graph* g, Node* input, const string& errmsg) { Node* ret; TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Error") |