aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/cc/ops
diff options
context:
space:
mode:
authorGravatar Peter Hawkins <phawkins@google.com>2017-07-07 06:08:53 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-07-07 06:13:19 -0700
commit7c2cfdef2693f17295471f717f86f31a3d308659 (patch)
treee7439eec9d86b164b5c7916e9ae6369f38d0daea /tensorflow/cc/ops
parent4e0f4e462f80e2dc84aa38da3fbc39cb15da3482 (diff)
Add Stack*V2 operators, which use a resource handle instead of a Ref tensor to store the stack handle.
Change in preparation for adding support for loop gradients in XLA-compiled graphs. XLA compilation requires the use of resource types instead of Ref types, and also requires statically known shapes. PiperOrigin-RevId: 161194280
Diffstat (limited to 'tensorflow/cc/ops')
-rw-r--r--tensorflow/cc/ops/op_gen_overrides.pbtxt4
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/cc/ops/op_gen_overrides.pbtxt b/tensorflow/cc/ops/op_gen_overrides.pbtxt
index 1dffb10c03..a1f79177f7 100644
--- a/tensorflow/cc/ops/op_gen_overrides.pbtxt
+++ b/tensorflow/cc/ops/op_gen_overrides.pbtxt
@@ -100,6 +100,10 @@ op { name: "Stack" skip: true }
op { name: "StackClose" skip: true }
op { name: "StackPop" skip: true }
op { name: "StackPush" skip: true }
+op { name: "StackV2" skip: true }
+op { name: "StackCloseV2" skip: true }
+op { name: "StackPopV2" skip: true }
+op { name: "StackPushV2" skip: true }
op { name: "TensorArrayCloseV2" skip: true }
op { name: "TensorArrayCloseV3" rename_to: "TensorArrayClose" }