Commit 60fc9f74 by Animesh Jain Committed by ziheng

[QNN] InferType changes that missed CI. (#3779)

parent f06ef4f2
......@@ -113,7 +113,10 @@ def concatenate(data,
# Find the dtype of the input expr. This is required for the requantize op. Since, this is
# concatenate op, the dtype of the input is same as dtype of the output.
data0 = relay.transform.infer_type(data[0])
mod = relay.Module.from_expr(data[0])
mod = relay.transform.InferType()(mod)
entry = mod["main"]
data0 = entry if isinstance(data[0], relay.Function) else entry.body
in_dtype = data0.checked_type.dtype
# First check if all the input qnn params match. If yes, we can call concatenate first, followed
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment