Skip to content

Commit

Permalink
Add comment
Browse files Browse the repository at this point in the history
  • Loading branch information
sfvaroglu committed Jan 21, 2022
1 parent feaed53 commit 246ea29
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions src/relay/qnn/op/rsqrt.cc
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,11 @@ Expr MakeQuantizedRsqrt(Expr x, Expr scale, Expr zero_point, Expr output_scale,
*/
Expr QnnRsqrtCanonicalize(const Attrs& attrs, const Array<Expr>& new_args,
const Array<tvm::relay::Type>& arg_types) {
// At this time, due to the complexity of implementing this op in int8 or uint8,
// we dequantize the input, run the op in float, and then quantize the output (as below).
// This acts as a placeholder for future hardware enablement, where more hardware specific
// canonicalization can be provided.

// Get the args.
QnnUnaryOpArguments args(new_args);

Expand Down

0 comments on commit 246ea29

Please sign in to comment.