Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 11 for nodedef (0.11 sec)

  1. tensorflow/c/kernels_test.cc

        NodeDef node_def;
        node_def.ParseFromArray(node_def_buf->data, node_def_buf->length);
        EXPECT_EQ(node_def.op(), "TestKernelAttrGetNodeDef");
        EXPECT_EQ(node_def.name(), "FakeNode");
        EXPECT_EQ(node_def.device(), "FakeDevice");
        EXPECT_EQ(node_def.attr_size(), 1);
        const ::tensorflow::AttrValue& value = node_def.attr().at("Attr");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

        // returned by _Retval nodes.
        std::unique_ptr<Graph> graph_;
    
        // Which device are these nodes on? Used to assign a device to the call
        // node.
        string device_;
    
        // NodeDef for the function call node.
        NodeDef call_node_def_;
    
        // Name that is used for the call node. This may not be
        // call_node_def_.name() if the client supplies a rewrite lambda.
        string function_def_name_;
    
    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/c/c_api_test.cc

      EXPECT_TRUE(neg == neg2);
      NodeDef node_def2;
      ASSERT_TRUE(GetNodeDef(neg2, &node_def2));
      EXPECT_EQ(node_def.DebugString(), node_def2.DebugString());
    
      TF_Operation* feed2 = TF_GraphOperationByName(graph, "feed");
      EXPECT_TRUE(feed == feed2);
      ASSERT_TRUE(GetNodeDef(feed, &node_def));
      ASSERT_TRUE(GetNodeDef(feed2, &node_def2));
      EXPECT_EQ(node_def.DebugString(), node_def2.DebugString());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_function_test.cc

        for (const NodeDef& node_def : fdef.node_def()) {
          for (int i = 0; i < node_def.input_size(); ++i) {
            const string& in = node_def.input(i);
            const auto& v =
                a_edges.insert({in, strings::StrCat(node_def.name(), ":", i)});
            ASSERT_TRUE(v.second) << "Duplicate edge " << in << " -> "
                                  << strings::StrCat(node_def.name(), ":", i)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

            op_count += self._count_op_with_name_and_attribute(
                nodes=func.node_def,
                op_name=op_name,
                attr_name=attr_name,
                attr_val=attr_val,
                get_op_name=get_op_name,
            )
        return op_count
    
      def _count_op_with_name_and_attribute(
          self,
          nodes: Iterable[node_def_pb2.NodeDef],
          op_name: str,
          attr_name: str,
          attr_val: _AttrValType,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      Scope root = Scope::NewRootScope().ExitOnError();
      {
        auto BuildNoopNode = [](absl::string_view name, Graph* graph) {
          NodeDefBuilder builder(name, "NoOp");
          NodeDef def;
          TF_CHECK_OK(builder.Finalize(&def));
    
          Status status;
          Node* node = graph->AddNode(def, &status);
          TF_CHECK_OK(status);
          return node;
        };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  7. pkg/scheduler/framework/plugins/defaultpreemption/default_preemption_test.go

    			},
    			nodes: []*v1.Node{
    				st.MakeNode().Name("node1").Capacity(nodeRes).Obj(),
    				st.MakeNode().Name("node2").Capacity(nodeRes).Obj(),
    				st.MakeNode().Name("node3").Capacity(nodeRes).Obj(),
    				st.MakeNode().Name("node4").Capacity(nodeRes).Obj(),
    			},
    			filteredNodesStatuses: framework.NodeToStatusMap{
    				"node1": framework.NewStatus(framework.Unschedulable),
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 31 15:52:16 UTC 2024
    - 82.8K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_test.cc

          "    signature {"
          "      name: 'FunctionWith__OP_NAME__'"
          "      output_arg {"
          "        name: 'out'"
          "        type: DT_STRING"
          "      }"
          "    }"
          "    node_def {"
          "      name: 'error_op'"
          "      op: '__OP_NAME__'"
          "    }"
          "    ret {"
          "      key: 'out'"
          "      value: 'error_op:out'"
          "    }";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  9. tensorflow/c/c_api.h

    // Changes an attr value in the node_def Protocol Buffer and sets a status upon
    // completion.
    TF_CAPI_EXPORT extern void TF_SetAttr(TF_Graph* graph, TF_Operation* op,
                                          const char* attr_name,
                                          TF_Buffer* attr_value_proto,
                                          TF_Status* status);
    
    // Clears the attr in the node_def Protocol Buffer and sets a status upon
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    #include "tensorflow/core/framework/bounds_check.h"
    #include "tensorflow/core/framework/graph_def_util.h"
    #include "tensorflow/core/framework/memory_types.h"
    #include "tensorflow/core/framework/node_def.pb.h"
    #include "tensorflow/core/framework/op_kernel.h"
    #include "tensorflow/core/framework/tensor.pb.h"
    #include "tensorflow/core/framework/types.h"
    #include "tensorflow/core/graph/algorithm.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top