Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for IRRewriter (0.65 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_to_vhlo.cc

    //   ==>
    //   result = op(V0)
    //   V1     = unrealized_cast(result)
    //   V2     = op2(V1)
    void ConvertAndWrapUsesInUnrealizedCast(Value result, TypeConverter &converter,
                                            IRRewriter &rewriter) {
      auto type = result.getType();
      result.setType(converter.convertType(result.getType()));
      auto new_value = converter.materializeArgumentConversion(
          rewriter, result.getLoc(), type, {result});
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 19:48:51 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc

    //    function.
    TF::PartitionedCallOp FinalizeFunctionRegister(
        PatternRewriter& rewriter, Value input, Value output,
        func::FuncOp& quantization_func, Operation* quantized_op,
        StringRef func_name, IRRewriter::InsertPoint original_point,
        Type quantize_result_type) {
      rewriter.create<func::ReturnOp>(input.getLoc(), ArrayRef<Value>({output}));
    
      quantization_func.setVisibility(func::FuncOp::Visibility::Private);
    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/lite/flatbuffer_export.cc

    }
    
    std::vector<std::pair<int, int>> Translator::ExtractControlEdges(
        mlir::Block* block) {
      std::vector<std::pair<int, int>> control_edges;
    
      mlir::IRRewriter rewriter(block->getParentOp()->getContext());
    
      // Since we're modifying *block, we store integer offsets to block->begin().
      llvm::DenseMap<Operation*, int> control_nodes_at;
      std::vector<Operation*> control_nodes;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top