experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix use of ISliceLayer

Browse Source
This commit is contained in:
Trevor Morris 2019-03-27 11:22:15 -07:00
parent 0a9195cc36
commit 0c39c30aeb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
View File

@ -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