masahi commented on a change in pull request #8309:
URL: https://github.com/apache/tvm/pull/8309#discussion_r657676668
##########
File path: src/relay/op/tensor/transform.cc
##########
@@ -477,19 +477,16 @@ Array<Array<Layout>> TransposeInferCorrectLayout(const
Attrs& attrs,
}
// Infer the output layout string based on the input layout and the axes.
+ Attrs new_attrs(params);
if (in_layout_str != "") {
for (auto axis : params->axes) {
ICHECK_LT(axis->value, in_layout_str.length());
out_layout_str += in_layout_str[axis->value];
}
- try {
- return Array<Array<Layout>>({{Layout(in_layout_str)},
{Layout(out_layout_str)}});
- } catch (const tvm::Error& e) {
- // If the layout string is invalid for any reason, give up.
Review comment:
ok restored try/catch for now.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
