- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for node_def (0.18 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)