Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc

        if (is_dynamic_broadcast) {
          auto conv_uses = (*conv_op.getODSResults(0).begin()).getUses();
          if (std::distance(conv_uses.begin(), conv_uses.end()) != 2 ||
              quant::FindUserOfType<shape::ShapeOfOp>(conv_op) == nullptr ||
              quant::FindUserOfType<mhlo::MulOp>(conv_op) == nullptr) {
            return rewriter.notifyMatchFailure(mul_op, [&](::mlir::Diagnostic
                                                               &diag) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 22:21:19 UTC 2024
    - 8.3K bytes
    - Viewed (0)
Back to top