Skip to content

Commit

Permalink
[QNN] InferType changes that missed CI. (apache#3779)
Browse files Browse the repository at this point in the history
  • Loading branch information
anijain2305 authored and wweic committed Aug 16, 2019
1 parent 5f29412 commit 79a8551
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion python/tvm/relay/qnn/op/qnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 79a8551

Please sign in to comment.