diff options
Diffstat (limited to 'tensorflow/core/kernels/session_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/session_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/session_ops.cc b/tensorflow/core/kernels/session_ops.cc index 4550115c19..59fb225b92 100644 --- a/tensorflow/core/kernels/session_ops.cc +++ b/tensorflow/core/kernels/session_ops.cc @@ -41,7 +41,7 @@ class GetSessionHandleOp : public OpKernel { : OpKernel(context) {} void Compute(OpKernelContext* ctx) override { - Tensor val = ctx->input(0); + const Tensor& val = ctx->input(0); int64 id = ctx->session_state()->GetNewId(); TensorStore::TensorAndKey tk{val, id, def().device()}; OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(def().name(), tk)); |