提交 c6a5c4b0 编写于 作者: N nhzlx

add comments for execute in ut_helper

上级 1bf9d9e9
......@@ -163,6 +163,9 @@ class TRTConvertValidation {
}
}
// We use the set 'neglected_output' here, because some Ops like batch norm,
// the outputs specified in the op des are only used during training,
// so we should neglect those output during inference.
void Execute(int batch_size,
std::unordered_set<std::string> neglected_output = {}) {
// Execute Fluid Op
......@@ -177,9 +180,6 @@ 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.
先完成此消息的编辑!
想要评论请 注册