Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for mutable_node (0.28 sec)

  1. tensorflow/cc/tools/freeze_saved_model.cc

    void GetNodeNameToNodeDefMap(
        GraphDef* graph_def,
        std::unordered_map<string, NodeDef*>* name_to_node_map) {
      for (size_t i = 0; i < graph_def->node_size(); i++) {
        NodeDef* node = graph_def->mutable_node(i);
        (*name_to_node_map)[node->name()] = node;
      }
    }
    
    // Strips off the tensor part of the tensor_name to get the node_name.
    const string GetNodeNameFromTensorName(string tensor_name) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 11 08:05:36 UTC 2023
    - 11.9K bytes
    - Viewed (0)
  2. tensorflow/cc/training/queue_runner_test.cc

      auto dequeue0 =
          QueueDequeue(root.WithOpName(kDequeueOp0), q0, {DataType::DT_FLOAT});
    
      GraphDef graph_def;
      TF_EXPECT_OK(root.ToGraphDef(&graph_def));
      for (auto& node : *graph_def.mutable_node()) {
        node.set_device("/cpu:0");
      }
      SessionOptions sess_options;
      sess_options.config.mutable_graph_options()->set_build_cost_model(1);
      std::unique_ptr<Session> session(NewSession(sess_options));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Sep 21 06:27:51 UTC 2019
    - 14.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_cluster_util.cc

    // ordered.
    void SortControlInputs(GraphDef* gdef) {
      int64_t num_nodes = gdef->node_size();
      for (int64_t i = 0; i < num_nodes; ++i) {
        NodeDef* node = gdef->mutable_node(i);
        // Stable sort control inputs and leave the order of data inputs unchanged.
        std::stable_sort(node->mutable_input()->begin(),
                         node->mutable_input()->end(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/saved_model/core/tf_concrete_function_loading_test.cc

          testing::ZeroArgInputSignature();
      *saved.mutable_output_signature() = testing::ZeroReturnOutputSignature();
      FunctionDef func = FuncDefWithNumInputsOutputs(0, 0);
      *func.mutable_signature()->mutable_name() = func_name;
    
      {
        std::unique_ptr<TFConcreteFunction> result;
        Status status =
            internal::LoadTFConcreteFunction(saved, &func, {}, context(), &result);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 19:16:58 UTC 2023
    - 10.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

        auto& attr = iter.second;
        if (attr.has_func()) {
          try_use_original_func_name(attr.mutable_func()->mutable_name());
        } else if (attr.has_list()) {
          for (auto& func_attr : *attr.mutable_list()->mutable_func()) {
            try_use_original_func_name(func_attr.mutable_name());
          }
        }
      }
    }
    
    Status Exporter::AddInstructionNode(Operation* inst) {
      std::unique_ptr<NodeDef> node_def;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

        auto& attr = iter.second;
        if (attr.has_func()) {
          try_use_original_func_name(attr.mutable_func()->mutable_name());
        } else if (attr.has_list()) {
          for (auto& func_attr : *attr.mutable_list()->mutable_func()) {
            try_use_original_func_name(func_attr.mutable_name());
          }
        }
      }
    }
    
    Status Exporter::AddInstructionNode(Operation* inst) {
      std::unique_ptr<NodeDef> node_def;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
Back to top