Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for FindUserOfType (0.14 sec)

  1. 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)
Back to top