Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for saveInsertionPoint (0.33 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

      // Note, 'inputs' and 'outputs' are the predetermined set of values that
      // should become the operands and return values, respectively.
      auto saved_insertion_point = builder.saveInsertionPoint();
      func::FuncOp new_func =
          CreateFnWithSignature(module, symbol_table, inputs, outputs, name);
    
      // This preserves the order of the ops that was in the original parent
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        // Create functions in a higher scope before restoring the insertion point.
        // Additionally, create the SymbolTable before further modifying the module.
        auto original_point = rewriter.saveInsertionPoint();
        rewriter.setInsertionPointAfter(op->getParentOfType<func::FuncOp>());
        SymbolTable manager(op->getParentOfType<ModuleOp>());
    
        // Constructs `then_branch`, which is executed when `if_cond` evaluates to
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      if (maybe_control_node == control_nodes->end()) {
        return op->getResults();
      }
      mlir::Region region;
      region.push_back(new mlir::Block);
      auto saved_pos = op_builder.saveInsertionPoint();
      op_builder.setInsertionPointToEnd(&region.front());
      mlir::Operation* cloned_op = op_builder.clone(*op);
      // Add the yield operation.
      op_builder.create<mlir::TFL::YieldOp>(op_loc, cloned_op->getResults());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
Back to top