- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for FindUserOfType (0.15 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
Operation* uniform_quantize_op; if (!has_i32_output) return op->getResult(0).getType(); if (fuse_bias_constant) { Operation* add_op = FindUserOfType<stablehlo::AddOp>(op); uniform_quantize_op = FindUserOfType<TFL::QuantizeOp>(add_op); } else { uniform_quantize_op = FindUserOfType<TFL::QuantizeOp>(op); } // StableHLO Quantizer outputs an i32 type. Rewrite to i8 type result // to meet TFLite op requirement.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0)