Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for subgraph_idx (0.25 sec)

  1. tensorflow/compiler/mlir/lite/experimental/common/outline_operations.h

      // Subgraphs are given a unique incremented integer id based on when
      // they were encountered in this pass.
      const int subgraph_id_;
    
      const llvm::StringRef dialect_namespace_;
    
      Subgraph(const llvm::SetVector<Operation*> partition_ops, int num_subgraphs)
          : partition_ops_(partition_ops),
            subgraph_id_(num_subgraphs),
            func_arguments_(AccumulateOperandsDefinedAbove(partition_ops)),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 17 18:49:43 UTC 2022
    - 6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc

          TypesFromValues(subgraph.FuncOutputs());
    
      FunctionType function_type =
          builder.getFunctionType(input_types, return_types);
    
      std::string function_name = absl::StrCat("func_", subgraph.subgraph_id_);
    
      func::FuncOp new_func = func::FuncOp::create(builder.getUnknownLoc(),
                                                   function_name, function_type);
      new_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
    - 8.5K bytes
    - Viewed (0)
Back to top