From 0c39c30aeb39d877a547c4dfb01d6d567c5ebbf0 Mon Sep 17 00:00:00 2001 From: Trevor Morris Date: Wed, 27 Mar 2019 11:22:15 -0700 Subject: [PATCH] Fix use of ISliceLayer --- tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc index a72702ea8ca..1d1b4e360cc 100644 --- a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc +++ b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc @@ -2355,7 +2355,7 @@ Status ConvertStridedSliceHelper(OpConverterParams* params, if (params->validation_only) return Status::OK(); nvinfer1::ISliceLayer* layer = params->converter->network()->addSlice( - input.tensor(), begin_dims, size_dims, stride_dims); + *input.tensor(), begin_dims, size_dims, stride_dims); params->outputs->push_back(TRT_TensorOrWeights(layer->getOutput(0))); return Status::OK(); #else