From bd64979fe95b6597123857d95b476ca98d0911a1 Mon Sep 17 00:00:00 2001 From: nhzlx Date: Tue, 24 Jul 2018 03:05:00 +0000 Subject: [PATCH] the argument should not be a const one --- paddle/fluid/inference/tensorrt/convert/fc_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/inference/tensorrt/convert/fc_op.cc b/paddle/fluid/inference/tensorrt/convert/fc_op.cc index c043c6aa3a4..498a3547de6 100644 --- a/paddle/fluid/inference/tensorrt/convert/fc_op.cc +++ b/paddle/fluid/inference/tensorrt/convert/fc_op.cc @@ -38,7 +38,7 @@ void Reorder2(nvinfer1::DimsHW shape, const T* idata, nvinfer1::DimsHW istrides, } // indata c * k // Reorder the data layout from CK to KC. -void ReorderCKtoKC(const TensorRTEngine::Weight& iweights, +void ReorderCKtoKC(TensorRTEngine::Weight& iweights, TensorRTEngine::Weight* oweights) { int c = iweights.dims[0]; int k = iweights.dims[1]; -- GitLab