Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
zhangjun committed Jan 20, 2022
2 parents 5ca9ee9 + 9b7db2b commit c030eb7
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions paddle/fluid/inference/tensorrt/op_teller.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1490,8 +1490,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
VLOG(3) << "the " << op_type
<< " does not have attr (keep_dim or dim or "
"reduce_all)";
std::cout << "attr " << desc.HasAttr("keep_dim") << " "
<< desc.HasAttr("dim") << " " << desc.HasAttr("reduce_all");
return false;
}

Expand All @@ -1510,7 +1508,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
std::vector<int32_t> dim =
BOOST_GET_CONST(std::vector<int32_t>, desc.GetAttr("dim"));
const auto input_shape = x_var_desc->GetShape();
std::cout << "paddle shape: " << input_shape.size() << std::endl;
for (auto x : dim) {
if (x == 0 || (x + input_shape.size() == 0)) return false;
}
Expand Down

0 comments on commit c030eb7

Please sign in to comment.