diff options
Diffstat (limited to 'tensorflow/docs_src/extend/adding_an_op.md')
-rw-r--r-- | tensorflow/docs_src/extend/adding_an_op.md | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/docs_src/extend/adding_an_op.md b/tensorflow/docs_src/extend/adding_an_op.md index c75c7f111d..f54f79cbf4 100644 --- a/tensorflow/docs_src/extend/adding_an_op.md +++ b/tensorflow/docs_src/extend/adding_an_op.md @@ -121,16 +121,16 @@ class ZeroOutOp : public OpKernel { Tensor* output_tensor = NULL; OP_REQUIRES_OK(context, context->allocate_output(0, input_tensor.shape(), &output_tensor)); - auto output = output_tensor->flat<int32>(); + auto output_flat = output_tensor->flat<int32>(); // Set all but the first element of the output tensor to 0. const int N = input.size(); for (int i = 1; i < N; i++) { - output(i) = 0; + output_flat(i) = 0; } // Preserve the first input value if possible. - if (N > 0) output(0) = input(0); + if (N > 0) output_flat(0) = input(0); } }; ``` |