提交 1bf9d9e9 编写于 作者: N nhzlx

fix comments

上级 324dd168
......@@ -177,6 +177,9 @@ class TRTConvertValidation {
ASSERT_FALSE(op_desc_->OutputArgumentNames().empty());
const size_t output_space_size = 3000;
for (const auto& output : op_desc_->OutputArgumentNames()) {
// Some Ops like batch norm, the output specified
// in the op des is only used during training, so we should
// neglect those ouput during inference.
if (neglected_output.count(output)) continue;
std::vector<float> fluid_out;
std::vector<float> trt_out(output_space_size);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册