Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for subgraph_name (0.19 sec)

  1. tensorflow/compiler/mlir/lite/tf_tfl_translate.cc

      for (auto *subgraph : *(*model)->subgraphs()) {
        std::string subgraph_name =
            subgraph->name() ? subgraph->name()->str() : "<<unnamed subgraph>>";
    
        std::cout << '\'' << subgraph_name << "' inputs:\n";
        int i = 0;
        for (auto input : *subgraph->inputs())
          print_buffer(*subgraph, i++, input, nullptr);
    
        std::cout << '\'' << subgraph_name << "' outputs:\n";
        mlir::Operation *terminator = nullptr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 14K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

    Status Encapsulator::Subgraph::MakeSequencingNode(const string& subgraph_name,
                                                      Graph* graph_out) {
      if (sequencer_ == nullptr) {
        NodeDef seq_def;
        // TODO(shikharagarwal): What source node should we use for errors?
        NodeDefBuilder builder(absl::StrCat(subgraph_name, "_sequencer"), "NoOp");
        builder.Attr(kXlaHostTransferSequencerAttr, subgraph_name);
        builder.Device(device_);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

    // have them, so we generate a name for subgraphs that are missing one here.
    // Note: in TFLite, the first subgraph is the entry point, and in MLIR that
    // represents TFLite, this entry point must be called "main"
    std::string SubgraphName(bool set_implicit_main_func, unsigned index,
                             const tflite::SubGraphT& subgraph) {
      if (index == 0 && set_implicit_main_func) {
        return "main";
      }
      if (subgraph.name.empty()) {
    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