Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for node_def (0.18 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/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)
  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/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)
  5. 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)
  6. 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)
  7. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      FunctionDefLibrary flib_def;
      FunctionDef func = FunctionDefHelper::Create(
          /*function_name=*/"Stateful_func", /*in_def=*/{},
          /*out_def=*/{"out: int32"},
          /*attr_def*/
          {}, /*node_def=*/
          {FunctionDefHelper::Const("shape_shape", 2),
           FunctionDefHelper::Const("minval", 1),
           FunctionDefHelper::Const("maxval", 20),
           {{"shape"},
            "RandomUniformInt",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  8. 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)
  9. 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)
  10. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

        # Create a new scope in case the local variables are leaked.
        self.symbol_table.enter_scope(scf_scope=True)
    
        # Create the iteration variable with index type
        assert len(body_def.args.args) == 1
        it_name = body_def.args.args[0].id
        it = self._ssa_name(it_name)
        self.symbol_table.insert_symbol(it_name, it, TFRTypes.INDEX)
    
        self.emit('scf.for {} = {} to {} step {} '.format(it, range_[0], range_[1],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
Back to top