Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 60 of 68 for GetOperands (0.21 sec)

  1. tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc

        : region_(&region) {
      if (region.empty()) return;
    
      assert(llvm::hasSingleElement(region.getBlocks()));
    
      auto results = region.front().getTerminator()->getOperands();
      if (results.empty()) return;
    
      backtracked_values_.reserve(results.size());
      for (auto result : results)
        backtracked_values_.push_back(backtrack_analysis.BacktrackValue(result));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc

      // UniformQuantized Ops are considered legal if its qint operands and
      // results are connected to TF CastOp.
      return op && llvm::all_of(op->getResults(), IsQintValueQintToIntCast) &&
             llvm::all_of(op->getOperands(), IsQintValueDefinedByIntToQintCast);
    }
    
    bool IsCastOpLegal(TF::CastOp cast_op) {
      // Consider qint <-> qint casts illegal.
      if (IsIllegalType(cast_op.getSrcT()) && IsIllegalType(cast_op.getDstT())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc

        func_metadata.original_name = func_op.getName();
        func_metadata.insertion_point = ++Block::iterator(func_op);
        func_metadata.ops.push_back(op);
    
        for (Value value : op->getOperands()) {
          std::string value_device = "";
    
          // If the value is defined as an argument of the func_op, adds it to
          // the argument list of the function that uses this op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 13.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc

          }
    
          write_op.erase();
        }
      }
    
      const int64_t num_results_before = function.getNumResults();
      auto return_operands = llvm::to_vector<4>(return_op.getOperands());
      auto result_types = llvm::to_vector<4>(return_op.getOperandTypes());
      llvm::SmallVector<std::pair<int64_t, llvm::StringRef>, 4>
          output_only_resources;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc

        const OpSetVector& cluster_dependent_ops) {
      Operation* other_op = nullptr;
      op->walk([&](Operation* inner_op) {
        ValueRange values = incoming ? ValueRange(inner_op->getOperands())
                                     : ValueRange(inner_op->getResults());
        llvm::SmallVector<Operation*, 4> candidates;
        for (Value value : values) {
          if (incoming) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 39.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc

      // so this op can be quantized. This is only applied on the returned result
      // because the error will not be accumulated.
    
      func.walk([&](func::ReturnOp ret) {
        int i = 0;
        for (Value returned : ret.getOperands()) {
          llvm::SmallVector<Value, 4> quantized;
          for (auto user : returned.getUsers()) {
            if (auto q = Quantized(user)) {
              quantized.push_back(q);
            }
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc

      // so this op can be quantized. This is only applied on the returned result
      // because the error will not be accumulated.
    
      func.walk([&](func::ReturnOp ret) {
        int i = 0;
        for (Value returned : ret.getOperands()) {
          llvm::SmallVector<Value, 4> quantized;
          for (auto user : returned.getUsers()) {
            if (auto q = Quantized(user)) {
              quantized.push_back(q);
            }
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc

    }
    // Gets the predecessors of an op wrapped in tf_executor.island.
    llvm::SmallVector<Operation*> GetPredecessors(Operation* op) {
      llvm::SmallVector<Operation*> predecessors;
      for (auto operand : op->getOperands()) {
        if (Operation* pred = operand.getDefiningOp()) {
          pred->walk([&](mlir::Operation* opinexecutor) {
            predecessors.push_back(opinexecutor);
          });
        }
      }
      return predecessors;
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 06:51:01 UTC 2024
    - 21.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

          // inputs.
          SmallVector<Value, 4> inputs;
          inputs.reserve(quantizing_op->getNumOperands());
          for (auto operand : quantizing_op->getOperands()) {
            Type operand_type = operand.getType();
            if (operand_type.isa<NoneType>()) {
              inputs.push_back(operand);
              continue;
            }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

        std::vector<tensorflow::Tensor>& tensors,
        std::vector<tensorflow::TensorValue>& inputs) {
      // Prepare the list of Tensor inputs for the kernel.
      for (auto it : llvm::enumerate(op_->getOperands())) {
        Value operand = it.value();
        size_t idx = it.index();
    
        tensorflow::XlaExpression expr = GetExprForOperand(operand, op_, idx);
        tensorflow::XlaExpression::Kind kind = expr.kind();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
Back to top