fix a bug

NHZlX · NHZlX · commit 4f71a3b12bfb · 2018-07-26T04:33:08.000Z
diff --git a/paddle/fluid/inference/tensorrt/convert/ut_helper.h b/paddle/fluid/inference/tensorrt/convert/ut_helper.h
@@ -162,7 +162,8 @@ class TRTConvertValidation {
 
       size_t fluid_out_size = fluid_out.size();
       if (if_add_batch_ == true) {
-        fluid_out_size = batch_size * (tensor->dims().size() / max_batch_size_);
+        fluid_out_size =
+            batch_size * (framework::product(tensor->dims()) / max_batch_size_);
       }
       // Compare two output
       ASSERT_FALSE(fluid_out.empty());

Original file line number	Diff line number	Diff line change
`@@ -162,7 +162,8 @@ class TRTConvertValidation {`
`162`	`162`
`163`	`163`	`size_t fluid_out_size = fluid_out.size();`
`164`	`164`	`if (if_add_batch_ == true) {`
`165`		`- fluid_out_size = batch_size * (tensor->dims().size() / max_batch_size_);`
	`165`	`+ fluid_out_size =`
	`166`	`+ batch_size * (framework::product(tensor->dims()) / max_batch_size_);`
`166`	`167`	`}`
`167`	`168`	`// Compare two output`
`168`	`169`	`ASSERT_FALSE(fluid_out.empty());`