opened 08:31AM - 16 Dec 21 UTC
JIT traced maskrcnn model from model zoo and provided desired layouts but couldn…'t avoid Attribute error. Any workarounds for this?
```
mod, params = relay.frontend.from_pytorch(script_module, shape_list)
mod = relay.transform.InferType()(mod)
mod["main"] = bind_params_by_name(mod["main"], params)
mod = transform.RemoveUnusedFunctions()(mod)
# no warnings when included these in desired layouts
#desired_layouts = {'nn.conv2d': ['NHWC', 'default'], 'image.resize2d': ['NHWC'], 'vision.roi_align': ['NHWC','default'], 'nn.conv2d_transpose':['NHWC','default']}
desired_layouts = {'nn.conv2d': ['NHWC', 'default']}
seq = tvm.transform.Sequential([relay.transform.RemoveUnusedFunctions(),
relay.transform.ConvertLayout(desired_layouts),
relay.transform.FoldConstant()])
with tvm.transform.PassContext(opt_level=3):
mod = seq(mod)
mod = relay.transform.DynamicToStatic()(mod)
# getting below mentioned here even after using DynamicToStatic()
mod = partition_for_vitis_ai(mod, params, dpu=target)
print(mod['main'])
```
`AttributeError: <class 'tvm.tir.expr.Any'> has no attribute value`
@jornt-xilinx Any suggestion on this?