Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for GetOperands (0.14 sec)

  1. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

          TF::TensorListSetItemOp op, OpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const {
        Location loc = op.getLoc();
        Value input = adaptor.getOperands()[0];
        Value index = adaptor.getOperands()[1];
        Value item = adaptor.getOperands()[2];
    
        IntegerType shape_dtype = rewriter.getIntegerType(32);
        auto item_rank = rewriter.create<TF::RankOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/deadness_analysis.cc

        if (common_inner_operands.empty()) {
          common_inner_operands.insert(common_inner_operands.end(),
                                       op->GetOperands().begin(),
                                       op->GetOperands().end());
        } else {
          common_inner_operands.clear();
          absl::c_copy_if(op->GetOperands(),
                          std::back_inserter(common_inner_operands),
                          [&](Predicate* sub_op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        Operation::operand_range mean(fused_batch_norm->getOperands());
        ::mlir::FloatAttr exponential_avg_factor;
        ::mlir::TF::FusedBatchNormV3Op root;
        Operation::operand_range offset(fused_batch_norm->getOperands());
        Operation::operand_range x(fused_batch_norm->getOperands());
        Operation::operand_range scale(fused_batch_norm->getOperands());
        Operation::operand_range variance(fused_batch_norm->getOperands());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

        Block& front = region->front();
        auto old_return = front.getTerminator();
        assert(old_return->getNumOperands() == op_->getNumResults());
        auto new_return_operands = llvm::to_vector<4>(old_return->getOperands());
        new_return_operands.resize(num_new_results_);
    
        // initialize return values for written resources to be the hoisted reads.
        for (Value resource : written_resources_) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

        }
      }
      return false;
    }
    
    bool HasDynamicExternalValues(Operation* op) {
      return op
          ->walk([](Operation* walked_op) {
            for (Value v : walked_op->getOperands()) {
              if (mlir::TF::CanBeRefined(v.getType())) {
                return WalkResult::interrupt();
              }
            }
            return WalkResult::advance();
          })
          .wasInterrupted();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

      llvm::SetVector<Operation*> new_ops;
    
      while (!ops_to_process.empty()) {
        for (Operation* op : ops_to_process) {
          if (predecessors) {
            for (Value operand : op->getOperands()) {
              // Stop at the block boundary.
              if (mlir::isa<BlockArgument>(operand)) continue;
    
              Operation* predecessor = operand.getDefiningOp();
              if (!operations->contains(predecessor) &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      mlir::Operation& return_op = region.back().back();
      mlir::Location loc = return_op.getLoc();
      op_builder.setInsertionPointToEnd(&region.back());
      op_builder.create<mlir::stablehlo::ReturnOp>(loc, return_op.getOperands());
      return_op.erase();
    }
    
    void InlineVhloOpRegion(mlir::Region& region, mlir::func::FuncOp func) {
      OpBuilder op_builder{region};
      mlir::IRMapping mapper;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc

            output_uniform_quantized_tensor_type};
        auto new_conv_op_with_output_type =
            rewriter.create<stablehlo::ConvolutionOp>(
                op.getLoc(), new_conv_output_types, op.getOperands(),
                op->getAttrs());
    
        rewriter.replaceAllUsesWith(op.getResult(),
                                    new_conv_op_with_output_type.getResult());
    
        auto new_output_dequant_op =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 64.6K bytes
    - Viewed (0)
Back to top