Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for QuantizeAvgPoolOpPattern (0.3 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc 

    // TODO(b/229183248): Remove this workaround after int8 kernels have been
// added to TF and XLA.
struct QuantizeAvgPoolOpPattern
    : public OpRewritePattern<quantfork::StorageCastOp> {
  explicit QuantizeAvgPoolOpPattern(MLIRContext* context)
      : OpRewritePattern<quantfork::StorageCastOp>(context, /*benefit=*/100) {}

  LogicalResult matchAndRewrite(quantfork::StorageCastOp sc_op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 23.6K bytes
    - Viewed (0)
Back to top