[Paddle-TRT] support batch axis concatenation when using dynamic shape (#31627)

* support batch axis concatenation when using dynamic shape

* opteller can't return true early, or some test will not be executed
test_benchmark_ci
zlsh80826 4 years ago committed by GitHub
parent d4282ea97e
commit 4ea3427865
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -159,7 +159,11 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
return false;
} else {
int axis = BOOST_GET_CONST(int, desc.GetAttr("axis"));
if (axis <= 0) return false;
if (with_dynamic_shape) {
if (axis < 0) return false;
} else {
if (axis <= 0) return false;
}
}
}
if (op_type == "transpose2" || op_type == "transpose") {

Loading…
Cancel
Save