Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 83 for getDefiningOp (0.15 sec)

  1. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc

      auto bias_op = op->getOperand(bias_index).getDefiningOp<arith::ConstantOp>();
      if (!affine_op || !bias_op || input_indices.size() != 2) return false;
      if (!mlir::isa<DenseFPElementsAttr>(bias_op.getValue())) return false;
      filter_index = affine_op.GetAffineOperandIndex();
      if (!op->getOperand(filter_index).getDefiningOp<arith::ConstantOp>()) {
        return false;
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 38.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc

          return;
        }
    
        // Invariant:
        // isa<quantfork::QuantizeCastOp>(dq_arg.getDefiningOp()) -->
        // getdq_arg.getType() != q_op.getResult().getType()
        //
        // as otherwise qdq pair would have been optimized away.
        auto qd_arg_def_q_op =
            dyn_cast_or_null<quantfork::QuantizeCastOp>(dq_arg.getDefiningOp());
        if (!qd_arg_def_q_op) {
          return;
        }
    
        qd_arg_def_q_op.emitWarning()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.cc

        SmallVector<Value, 1> new_values;
        bool should_rewrite = false;
        for (auto operand : op.getArgs()) {
          if (auto corert_const_dense_tensor_op =
                  operand.getDefiningOp<corert::ConstDenseTensorOp>()) {
            new_values.push_back(
                rewriter.create<fallback_async::ConstDenseTensorOp>(
                    op.getLoc(), rewriter.getType<fallback::TFTensorType>(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 15.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/dilated_conv.h

      }
    
      // Check if the ConvOp's input is defined by `Expand` op, and the output used
      // by `Squeeze` op.
      Operation* producer_op = op.getOperand(0).getDefiningOp();
      if (!producer_op || producer_op->getNumResults() != 1) {
        return rewriter.notifyMatchFailure(
            op, "op doesn't have a producer node that has a single result");
      }
      if (!producer_op->hasOneUse() ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc

                  preceding_op)) {
            if (auto q_op = llvm::dyn_cast_or_null<quantfork::QuantizeCastOp>(
                    dq_op.getArg().getDefiningOp())) {
              Operation* q_op_input = q_op.getArg().getDefiningOp();
              is_weight_constant =
                  q_op_input && q_op_input->hasTrait<OpTrait::ConstantLike>();
            }
          }
    
          if (!is_weight_constant) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 16.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc

                                         ShapedType result_type,
                                         StringRef func_name,
                                         Value& func_input_arg) {
      Operation* input_op = input_val.getDefiningOp();
    
      Operation* insertion_point = input_op->getParentOfType<func::FuncOp>();
      if (!insertion_point) insertion_point = input_op->getParentOfType<ModuleOp>();
      rewriter.setInsertionPointAfter(insertion_point);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h

          // The input of the quantize op has already been quantized, i.e.
          // rescale.
          return failure();
        }
    
        Operation* operand_op = operand.getDefiningOp();
        if (operand_op == nullptr) {
          // When `QuantizeOpT`'s operand does not have a defining op, it means it
          // is a `BlockArgument`. The pattern does not match if there is no op to
          // quantize.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_to_vhlo.cc

      auto new_value = converter.materializeArgumentConversion(
          rewriter, result.getLoc(), type, {result});
      rewriter.replaceAllUsesExcept(result, new_value, new_value.getDefiningOp());
    }
    
    // Wrap operands in an an unrealized cast to create a cast to buffer any type
    // changes to the operand, and apply type converter to operands:
    //   V0 = op(operand)
    //   ==>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 19:48:51 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc

          return;
        }
    
        // Invariant:
        // isa<quantfork::QuantizeCastOp>(dq_arg.getDefiningOp()) -->
        // getdq_arg.getType() != q_op.getResult().getType()
        //
        // as otherwise qdq pair would have been optimized away.
        auto qd_arg_def_q_op =
            dyn_cast_or_null<quantfork::QuantizeCastOp>(dq_arg.getDefiningOp());
        if (!qd_arg_def_q_op) {
          return;
        }
    
        qd_arg_def_q_op.emitWarning()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h

    template <typename T = Operation*>
    Operation* FindOperandOfType(Operation* op) {
      for (Value operand_value : op->getOperands()) {
        if (isa<T>(operand_value.getDefiningOp())) {
          return operand_value.getDefiningOp();
        }
      }
      return nullptr;
    }
    
    // Returns the function attribute for the given call op which is lifted for
    // quantization.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.9K bytes
    - Viewed (0)
Back to top