Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 77 for getDefiningOp (0.18 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h

      // which is cyclical.
      return success();
    }
    
    inline OpFoldResult foldIdempotent(Operation* op) {
      if (op->getNumOperands() == 1) {
        auto* argumentOp = op->getOperand(0).getDefiningOp();
        if (argumentOp && op->getName() == argumentOp->getName()) {
          // Replace the outer operation output with the inner operation.
          return op->getOperand(0);
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc

                  extern_value.getDefiningOp())) {
            extern_values.insert(extern_value);
            continue;
          }
          // Add constant at start of region.
          auto const_builder =
              OpBuilder(&it.value()->front(), it.value()->front().begin());
          auto const_value = const_builder.clone(*extern_value.getDefiningOp());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc

            if (!element_type.isF32()) {
              continue;
            }
    
            // Skip when there is any already existing CustomAggregatorOp found.
            Operation *defining_op = input.get().getDefiningOp();
            if (dyn_cast_or_null<TF::CustomAggregatorOp>(defining_op)) {
              continue;
            }
    
            // Skip calibration when the given operand comes from a constant.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 14.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc

           llvm::zip(cluster_terminator->getOperands(), cluster.getResults())) {
        Value cluster_terminator_operand = std::get<0>(result);
        if (cluster_terminator_operand.getDefiningOp() &&
            cluster.getOperation()->isProperAncestor(
                cluster_terminator_operand.getDefiningOp())) {
          new_cluster_results.push_back(cluster_terminator_operand);
          new_cluster_result_types.push_back(cluster_terminator_operand.getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.6K bytes
    - Viewed (0)
Back to top