Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for subgraph_name (0.16 sec)

  1. 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)
  2. 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